Update app.py

app.py

@@ -6,18 +6,12 @@ import logging
 logging.basicConfig(level=logging.INFO)
 
 # Load tokenizer and model
-model_name = "mrm8488/t5-base-finetuned-wikiSQL"  # Alternative model
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-
-def preprocess_query(query):
-    # Example preprocessing: convert to lowercase
-    return query.lower()
+tokenizer = AutoTokenizer.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
+model = AutoModelForSeq2SeqLM.from_pretrained("hrshtsharma2012/NL2SQL-Picard-final")
 
 def generate_sql(query):
     try:
-        processed_query = preprocess_query(query)
-        inputs = tokenizer(processed_query, return_tensors="pt", padding=True)
+        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
@@ -30,7 +24,7 @@ interface = gr.Interface(
     fn=generate_sql,
     inputs=gr.Textbox(lines=2, placeholder="Enter your natural language query here..."),
     outputs="text",
-    title="NL to SQL with T5",
+    title="NL to SQL with Picard",
     description="This model converts natural language queries into SQL. Enter your query!"
 )