Update app.py

app.py

@@ -3,6 +3,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers import TapexTokenizer, BartForConditionalGeneration
 import pandas as pd
 import torch
+
+import numpy as np
+import time
+import os
 #import pkg_resources
 
 '''
@@ -42,8 +46,8 @@ bot_input_ids = None
 
 def chatbot_response(user_message):
 
-    global new_chat
     global chat_history_ids
+    global bot_input_ids
     # Check if the user input is a question
     is_question = "?" in user_message
 
@@ -61,7 +65,7 @@ def chatbot_response(user_message):
         response = chatbot_tokenizer.decode(outputs[0], skip_special_tokens=True)
         '''
         # encode the new user input, add the eos_token and return a tensor in Pytorch
-        new_user_input_ids = chatbot_tokenizer.encode("User: " + user_message + chatbot_tokenizer.eos_token, return_tensors='pt')
+        new_user_input_ids = chatbot_tokenizer.encode(user_message + chatbot_tokenizer.eos_token, return_tensors='pt')
     
         # append the new user input tokens to the chat history
         if chat_history_ids is not None: