switch back to 90M model

main.py

@@ -1,7 +1,7 @@
 from dotenv import load_dotenv
 import os
 
-from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
+from transformers import BlenderbotSmallForConditionalGeneration, AutoTokenizer
 from helpers import in_cache
 
 from fastapi import FastAPI
@@ -13,8 +13,8 @@ from starlette.middleware.cors import CORSMiddleware
 
 load_dotenv()
 
-model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill", cache_dir="new_cache_dir/")
-tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill", cache_dir="new_cache_dir/")
+model = BlenderbotSmallForConditionalGeneration.from_pretrained("facebook/blenderbot_small-90M", cache_dir="new_cache_dir/")
+tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot_small-90M", cache_dir="new_cache_dir/")
 class UserMSGRequest(BaseModel):
     message: str