Upload demo/hf_repo_assets.py with huggingface_hub

demo/hf_repo_assets.py

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