fix bug

webui/app.py

@@ -24,7 +24,7 @@ app.mount("/static", StaticFiles(directory=str(WEBUI_DIR / "static")), name="sta
 app.mount("/results", StaticFiles(directory=str(RESULT_DIR)), name="results")
 
 # ---- weights repo ----
-WEIGHTS_REPO = os.getenv("TASKCLIP_WEIGHTS_REPO", "BiasLab2025/YOUR-WEIGHTS-REPO")  # <-- change default
+WEIGHTS_REPO = os.getenv("TASKCLIP_WEIGHTS_REPO", "BiasLab2025/taskclip-weights")
 WEIGHTS_DIR = get_weights_dir(WEIGHTS_REPO)
 
 CKPT_DIR = WEIGHTS_DIR / "checkpoints"

webui/weights.py

@@ -1,12 +1,19 @@
-# webui/weights.py
 import os
 from pathlib import Path
 from huggingface_hub import snapshot_download
 
 def get_weights_dir(repo_id: str) -> Path:
-    token = os.getenv("HF_TOKEN")  # only needed if repo is private
+    # repo_id must be like "BiasLab2025/taskclip-weights" (NOT a URL)
+    repo_id = repo_id.strip()
+    if repo_id.startswith("http"):
+        # allow passing a full URL by accident
+        repo_id = repo_id.rstrip("/").split("huggingface.co/")[-1]
+
+    token = os.getenv("HF_TOKEN")  # only needed if the repo is private
+
     p = snapshot_download(
         repo_id=repo_id,
+        repo_type="model",          # IMPORTANT for your weights repo
         local_dir="weights_cache",
         local_dir_use_symlinks=False,
         token=token,