Update app.py

app.py

@@ -1,11 +1,13 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
+from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+
+
 import torch
 
-chat_tkn = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-mdl = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
+chat_tkn = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+mdl = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 
 def converse(user_input, chat_history=[]):
-    # tokenize the new input sentence
+    #tkn_ids = chat_tkn(input(">> User:") + chat_tkn.eos_token, return_tensors='pt')
     user_input_ids = chat_tkn(input + chat_tkn.eos_token, return_tensors='pt')
 
     # append the new user input tokens to the chat history