Update app.py

app.py

@@ -10,7 +10,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained("hrshtsharma2012/NL2SQL-Picard-fin
 spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
-    inputs = tokenizer(query, return_tensors="pt", padding=True)
+    # Add a prefix to prompt the model to generate SQL
+    prefixed_query = f"Translate to SQL: {query}"
+    inputs = tokenizer(prefixed_query, return_tensors="pt", padding=True)
     outputs = model.generate(**inputs, max_length=512)
     sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return sql_query