Update app.py

app.py

@@ -1,15 +1,23 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import logging
+
+# Set up logging
+logging.basicConfig(level=logging.INFO)
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
 model = AutoModelForSeq2SeqLM.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
 
 def generate_sql(query):
-    inputs = tokenizer(query, return_tensors="pt", padding=True)
-    outputs = model.generate(**inputs, max_length=512)
-    sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return sql_query
+    try:
+        inputs = tokenizer(query, return_tensors="pt", padding=True)
+        outputs = model.generate(**inputs, max_length=512)
+        sql_query = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        return sql_query
+    except Exception as e:
+        logging.error(f"Error generating SQL query: {e}")
+        return "Error generating SQL query"
 
 # Create a Gradio interface
 interface = gr.Interface(