Update app.py

app.py

@@ -13,7 +13,11 @@ nl2sql_pipeline = pipeline("text2text-generation", model=model, tokenizer=tokeni
 spider_dataset = load_dataset("spider", split='train[:5]')
 
 def generate_sql(query):
+    # Ensure the query is a string
+    query = str(query)
+    # Run the pipeline
     results = nl2sql_pipeline(query)
+    # Extract the SQL query
     sql_query = results[0]['generated_text']
     return sql_query
 
@@ -31,5 +35,5 @@ interface = gr.Interface(
 )
 
 # Launch the app
-if __name__ == "__main__": 
+if __name__ == "__main__":
     interface.launch()