add cache directory

main.py

@@ -13,8 +13,8 @@ from starlette.middleware.cors import CORSMiddleware
 
 load_dotenv()
 
-model = BlenderbotSmallForConditionalGeneration.from_pretrained("facebook/blenderbot_small-90M")
-tokenizer = AutoTokenizer.from_pretrained("facebook/blenderbot_small-90M")
+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