Update app.py

app.py

@@ -13,10 +13,12 @@ nl2sql_pipeline = pipeline("text2text-generation", model=model, tokenizer=tokeni
 spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
-    results = nl2sql_pipeline(query)
+    formatted_query = f"translate English to SQL: {query}"
+    results = nl2sql_pipeline(formatted_query)
     sql_query = results[0]['generated_text']
     return sql_query
 
+
 # Use examples from the Spider dataset
 example_questions = [(question['question'],) for question in spider_dataset]