Update app.py

app.py

@@ -45,7 +45,7 @@ def chatbot_response(user_message):
 
     global bot_input_ids
     # encode the new user input, add the eos_token and return a tensor in Pytorch
-    new_user_input_ids = chatbot_tokenizer.encode(user_message + 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 bot_input_ids is None:
@@ -57,7 +57,7 @@ def chatbot_response(user_message):
     chat_history_ids = chatbot_model.generate(bot_input_ids, max_length=1000, pad_token_id=tokenizer.eos_token_id)
 
     # pretty print last ouput tokens from bot
-    response = "DialoGPT: {}".format(chatbot_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True))
+    response = chatbot_tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
 
     return response