Update app.py

app.py

@@ -5,7 +5,7 @@ title = "Have Fun With ChubbyBot"
 description = """
 <p>
 <center>
-The bot is trained on blended_skill_talk dataset using facebook/blenderbot-400M-distill.
+The bot is trained on blended_skill_talk dataset using facebook/blenderbot-1B-distill.
 <img src="https://huggingface.co/spaces/EXFINITE/BlenderBot-UI/resolve/main/img/cover.png" alt="rick" width="250"/>
 </center>
 </p>
@@ -15,8 +15,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1907.066
 import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, BlenderbotForConditionalGeneration, BlenderbotForCausalLM, BlenderbotTokenizer
 
-tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill",add_cross_attention=False)
+tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-1B-distill")
+model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-1B-distill",add_cross_attention=False)
 
 def predict(input, history=[]):
     # tokenize the new input sentence