Update app.py

app.py

@@ -11,9 +11,9 @@ def generate_sql(query):
     outputs = model.generate(**inputs, max_length=512)
     sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # Check if the output is a valid SQL query
-    if not sql_query.lower().startswith("select"):
-        return "The model did not generate a valid SQL query. Please try a different input or use a different model."
+    # Check if the output is the same as the input
+    if sql_query.strip().lower() == query.strip().lower():
+        return "The model did not generate a SQL query. Please try a different input or use a different model."
     
     return sql_query