Upload demo/hf_repo_assets.py with huggingface_hub

demo/hf_repo_assets.py

@@ -6,9 +6,11 @@ from pathlib import Path
 
 from huggingface_hub import snapshot_download
 
+DEFAULT_MODEL_REPO_ID = "SynLayers/Bbox-caption-8b"
 
-def get_stage2_model_repo_id() -> str | None:
-    return os.environ.get("SYNLAYERS_STAGE2_MODEL_REPO") or os.environ.get("SYNLAYERS_MODEL_REPO")
+
+def get_model_repo_id() -> str | None:
+    return os.environ.get("SYNLAYERS_MODEL_REPO") or DEFAULT_MODEL_REPO_ID
 
 
 def get_cache_dir() -> str | None:
@@ -17,8 +19,8 @@ def get_cache_dir() -> str | None:
 
 @lru_cache(maxsize=4)
 def ensure_repo_assets(repo_id: str | None = None) -> Path | None:
-    """Download Stage 2 runtime assets from the configured model repo."""
-    resolved_repo_id = repo_id or get_stage2_model_repo_id()
+    """Download required runtime assets from the uploaded model repo when configured."""
+    resolved_repo_id = repo_id or get_model_repo_id()
     if not resolved_repo_id:
         return None
 
@@ -41,7 +43,7 @@ def ensure_repo_assets(repo_id: str | None = None) -> Path | None:
 
 
 def build_repo_asset_overrides(repo_id: str | None = None) -> dict[str, str]:
-    """Return repo-local Stage 2 asset paths after downloading the uploaded bundle."""
+    """Return repo-local asset paths after downloading the uploaded bundle."""
     local_root = ensure_repo_assets(repo_id)
     if local_root is None:
         return {}