Update app.py

app.py

@@ -58,7 +58,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(input(">> User:") + chatbot_tokenizer.eos_token, return_tensors='pt')
+        new_user_input_ids = chatbot_tokenizer.encode("User: " + user_message + chatbot_tokenizer.eos_token, return_tensors='pt')
     
         # append the new user input tokens to the chat history
         bot_input_ids = torch.cat([chat_history_ids, new_user_input_ids], dim=-1) if new_chat is False else new_user_input_ids