Update app.py

app.py

@@ -14,7 +14,8 @@ spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
     results = nl2sql_pipeline(query)
-    sql_query = results[0]['generated_text']
+    # Ensure that we get the first generated text as the SQL query
+    sql_query = results[0]['generated_text'] if results else "No SQL query generated"
     return sql_query
 
 # Use examples from the Spider dataset
@@ -32,4 +33,4 @@ interface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()