Update app.py

app.py

@@ -13,10 +13,10 @@ The bot is trained on blended_skill_talk dataset using facebook/blenderbot-400M-
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1907.06616' target='_blank'>Recipes for building an open-domain chatbot</a></p><p style='text-align: center'><a href='https://parl.ai/projects/recipes/' target='_blank'>Original PARLAI Code</a></p></center></p>"
 
 import torch
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, BlenderbotForCausalLM, BlenderbotTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-model = AutoModelForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
+tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
+model = BlenderbotForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
 
 def predict(input, history=[]):
     # tokenize the new input sentence