Update app.py

app.py

@@ -1,16 +1,17 @@
 import gradio as gr
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import AutoModelForQuestionAnswering, AutoTokenizer
+from transformers import TapasTokenizer, TapasForQuestionAnswering
 
 def execute_sql(user_query):
     model_name = "microsoft/tapex-large-sql-execution"  # Tapex large SQL execution model
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+    model = AutoModelForQuestionAnswering.from_pretrained(model_name)
 
     inputs = tokenizer(user_query, return_tensors="pt", padding=True)
-    outputs = model.generate(inputs['input_ids'], attention_mask=inputs['attention_mask'], max_length=1024)
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    outputs = model(**inputs)
+    answer = tokenizer.decode(inputs['input_ids'][0][outputs['start_logits'].argmax():outputs['end_logits'].argmax() + 1])
 
-    return response
+    return answer
 
 '''
 def chatbot_response(user_message):