Update app.py

app.py

@@ -13,10 +13,10 @@ nl2sql_pipeline = pipeline("text2text-generation", model=model, tokenizer=tokeni
 spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
-    # Ensure the query is a string
-    query = str(query)
+    # Format the input for the model
+    input_text = f"translate English to SQL: {query}"
     # Run the pipeline
-    results = nl2sql_pipeline(query)
+    results = nl2sql_pipeline(input_text)
     # Extract the SQL query
     sql_query = results[0]['generated_text']
     return sql_query