Update app.py

app.py

@@ -1,50 +1,28 @@
 import gradio as gr
 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 = AutoModelForQuestionAnswering.from_pretrained(model_name)
+# Load the Tapas model and tokenizer
+model_name = "google/tapas-large-finetuned-wtq"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForQuestionAnswering.from_pretrained(model_name)
 
-    inputs = tokenizer(user_query, return_tensors="pt", padding=True)
+def execute_sql(user_query):
+    inputs = tokenizer(user_query, return_tensors="pt")
     outputs = model(**inputs)
     answer = tokenizer.decode(inputs['input_ids'][0][outputs['start_logits'].argmax():outputs['end_logits'].argmax() + 1])
-
     return answer
 
-'''
-def chatbot_response(user_message):
-    # Your chatbot code goes here (using GPT-2 or any other text generation model)
-    # For example, you can use the GPT-2 code from the previous responses
-
-    return chatbot_generated_response
-
-# Define the chatbot and SQL execution interface using Gradio
-chatbot_interface = gr.Interface(
-    fn=chatbot_response,
-    inputs=gr.Textbox(prompt="You:"),
-    outputs=gr.Textbox(),
-    live=True,
-    capture_session=True,
-    title="Chatbot",
-    description="Type your message in the box above, and the chatbot will respond.",
-)
-'''
-
-sql_execution_interface = gr.Interface(
+# Define the chatbot interface using Gradio
+iface = gr.Interface(
     fn=execute_sql,
-    inputs=gr.Textbox(prompt="Enter your SQL query:"),
+    inputs=gr.Textbox(prompt="Enter your question:"),
     outputs=gr.Textbox(),
     live=True,
     capture_session=True,
-    title="SQL Execution",
-    description="Type your SQL query in the box above, and the chatbot will execute it.",
+    title="Database Question Answering Chatbot",
+    description="Type your questions about the database in the box above, and the chatbot will provide answers.",
 )
 
-# Combine the chatbot and SQL execution interfaces
-#combined_interface = gr.Interface([chatbot_interface, sql_execution_interface], layout="horizontal")
-
-# Launch the combined Gradio interface
+# Launch the Gradio interface
 if __name__ == "__main__":
-    sql_execution_interface.launch()
+    iface.launch()