Update app.py

app.py

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