Update app.py

app.py

@@ -80,6 +80,15 @@ def chat(input, history=[]):
 
     else:
     '''
+
+    if is_question: 
+        chatbot_model_name = "microsoft/tapex-large-finetuned-wtq"
+        tokenizer = AutoTokenizer.from_pretrained(chatbot_model_name)
+        model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
+    else:
+        chatbot_model_name = "microsoft/DialoGPT-medium" 
+        tokenizer = AutoTokenizer.from_pretrained(chatbot_model_name)
+        model = AutoModelForCausalLM.from_pretrained(chatbot_model_name)
     
     # tokenize the new input sentence
     new_user_input_ids = tokenizer.encode(input + tokenizer.eos_token, return_tensors='pt')