Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from datasets import load_dataset
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
@@ -11,6 +10,11 @@ def generate_sql(query):
     inputs = tokenizer(input_text, return_tensors="pt", padding=True)
     outputs = model.generate(**inputs, max_length=512)
     sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # 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
 
 # Create a Gradio interface
@@ -19,7 +23,7 @@ interface = gr.Interface(
     inputs=gr.Textbox(lines=2, placeholder="Enter your natural language query here..."),
     outputs="text",
     title="NL to SQL with Picard",
-    description="This model converts natural language queries into SQL using the Spider dataset. Enter your query!"
+    description="This model converts natural language queries into SQL. Enter your query!"
 )
 
 # Launch the app