Update app.py

app.py

@@ -6,10 +6,12 @@ tokenizer = AutoTokenizer.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
 model = AutoModelForSeq2SeqLM.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
 
 def generate_sql(query):
-    input_text = "translate English to SQL: " + query
+    input_text = "SQL: " + query
     inputs = tokenizer(input_text, return_tensors="pt", padding=True)
     outputs = model.generate(**inputs, max_length=512)
     sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Remove the "SQL: " prefix from the output
+    sql_query = sql_query.replace("SQL: ", "")
     return sql_query
 
 # Create a Gradio interface
@@ -18,9 +20,10 @@ interface = gr.Interface(
     inputs=gr.Textbox(lines=2, placeholder="Enter your natural language query here..."),
     outputs="text",
     title="NL to SQL with Picard",
-    description="This model converts natural language queries into SQL using the Picard model. Enter your query!"
+    description="This model converts natural language queries into SQL. Enter your query!"
 )
 
 # Launch the app
 if __name__ == "__main__":
     interface.launch()
+