Update app.py

app.py

@@ -11,10 +11,12 @@ 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"Translate to SQL: {query}"
+    prefixed_query = f"T1.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)
+    # 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