Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, TapasForQuestionAnswering, TapasTokenizer
 
 # Load the models and tokenizers
-tapas_model_name = "google/tapas-large-finetuned-wtq"
+tapas_model_name = "microsoft/tapex-large-finetuned-wtq"
 dialogpt_model_name = "microsoft/DialoGPT-medium"
 
 tapas_tokenizer = TapasTokenizer.from_pretrained(tapas_model_name)
-tapas_model = TapasForQuestionAnswering.from_pretrained(tapas_model_name)
+tapas_model = BartForConditionalGeneration.from_pretrained(tapas_model_name)
 
 dialogpt_tokenizer = AutoTokenizer.from_pretrained(dialogpt_model_name)
-dialogpt_model = AutoModelForSeq2SeqLM.from_pretrained(dialogpt_model_name)
+dialogpt_model = AutoModelForSeqCausalLM.from_pretrained(dialogpt_model_name)
 
 def answer_table_question(table, question):
     encoding = tapas_tokenizer(table=table, query=question, return_tensors="pt")