update

app.py

@@ -14,6 +14,8 @@ from src.util.seeding import seed_all
 import logging
 from huggingface_hub import snapshot_download
 
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 seed = 1234
 seed_all(seed)
 if torch.cuda.is_available():
@@ -44,6 +46,7 @@ def get_demo():
         local_dir = snapshot_download(
             repo_id="GDAOSU/olbedo",
             allow_patterns=f"{selected_model}/*",
+            token=HF_TOKEN,
         )
 
         model_path = os.path.join(local_dir, selected_model)