app.py beta 2 (Push 2)

app.py

@@ -20,13 +20,14 @@ db = client["test"]
 products_collection = db["products"]
 
 # BlenderBot Setup
-model_name = "SyedHutter/blenderbot_model/blenderbot_model"  # Points to subdirectory
+model_repo = "SyedHutter/blenderbot_model"  # Repo ID
+model_subfolder = "blenderbot_model"        # Subdirectory within repo
 model_dir = "/home/user/app/blenderbot_model"
 
 if not os.path.exists(model_dir):
-    logger.info(f"Downloading {model_name} to {model_dir}...")
-    tokenizer = BlenderbotTokenizer.from_pretrained(model_name)
-    model = BlenderbotForConditionalGeneration.from_pretrained(model_name)
+    logger.info(f"Downloading {model_repo}/{model_subfolder} to {model_dir}...")
+    tokenizer = BlenderbotTokenizer.from_pretrained(model_repo, subfolder=model_subfolder)
+    model = BlenderbotForConditionalGeneration.from_pretrained(model_repo, subfolder=model_subfolder)
     os.makedirs(model_dir, exist_ok=True)
     tokenizer.save_pretrained(model_dir)
     model.save_pretrained(model_dir)