Update app.py

app.py

@@ -15,6 +15,8 @@ spider_dataset = load_dataset("spider", split='train[:5]')
 def generate_sql(query):
     results = nl2sql_pipeline(query)
     sql_query = results[0]['generated_text']
+    # Post-process the output to ensure it's a valid SQL query
+    sql_query = sql_query.replace('<pad>', '').replace('</s>', '').strip()
     return sql_query
 
 # Use examples from the Spider dataset