Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 from datasets import load_dataset
@@ -13,11 +14,7 @@ nl2sql_pipeline = pipeline("text2text-generation", model=model, tokenizer=tokeni
 spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
-    # Format the input for the model
-    input_text = f"translate English to SQL: {query}"
-    # Run the pipeline
-    results = nl2sql_pipeline(input_text)
-    # Extract the SQL query
+    results = nl2sql_pipeline(query)
     sql_query = results[0]['generated_text']
     return sql_query
 
@@ -36,4 +33,4 @@ interface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()