Update hf_space/app.py

hf_space/app.py

@@ -24,18 +24,68 @@ if str(ROOT_DIR) not in sys.path:
 print("[GestureLSM] sys.path:")
 for entry in sys.path:
     print("  ", entry)
-CKPT_DIR = BASE_DIR / "ckpt"
-CKPT_DIR.mkdir(parents=True, exist_ok=True)
+ASSET_CACHE_DIR = BASE_DIR / "asset_cache"
+ASSET_CACHE_DIR.mkdir(parents=True, exist_ok=True)
 
 weights_repo = os.environ.get("HF_GESTURELSM_WEIGHTS_REPO", "").strip()
 if weights_repo:
-    snapshot_download(
-        repo_id=weights_repo,
-        repo_type="dataset",
-        local_dir=CKPT_DIR,
-        local_dir_use_symlinks=False,
-        allow_patterns=["*.pth", "*.bin", "*.npz", "*.npy"],
+    downloaded_root = Path(
+        snapshot_download(
+            repo_id=weights_repo,
+            repo_type="dataset",
+            local_dir=ASSET_CACHE_DIR,
+            local_dir_use_symlinks=False,
+            allow_patterns=[
+                "*.pth",
+                "*.pt",
+                "*.bin",
+                "*.npz",
+                "*.npy",
+                "*.tar",
+                "*.tar.gz",
+                "*.zip",
+            ],
+        )
     )
+else:
+    downloaded_root = ASSET_CACHE_DIR
+
+
+def _sync_downloaded_assets(download_root: Path) -> None:
+    """Copy all downloaded asset files into the repo root."""
+
+    allowed_suffixes = {
+        ".pth",
+        ".pt",
+        ".bin",
+        ".npz",
+        ".npy",
+        ".tar",
+        ".gz",
+        ".zip",
+    }
+
+    copied = 0
+    for file_path in download_root.rglob("*"):
+        if not file_path.is_file():
+            continue
+        if file_path.suffix.lower() not in allowed_suffixes:
+            continue
+        relative = file_path.relative_to(download_root)
+        destination = ROOT_DIR / relative
+        destination.parent.mkdir(parents=True, exist_ok=True)
+        try:
+            shutil.copy2(file_path, destination)
+            copied += 1
+        except Exception as exc:  # pragma: no cover - defensive
+            print(f"[GestureLSM] Failed to copy {file_path} -> {destination}: {exc}")
+
+    if copied:
+        print(f"[GestureLSM] Synced {copied} asset files from {download_root} to repository root")
+
+
+
+_sync_downloaded_assets(downloaded_root)
 
 # Ensure expected runtime directories exist so the demo can write outputs.
 datasets_hub_dir = ROOT_DIR / "datasets" / "hub"
@@ -46,7 +96,7 @@ smplx_dest = datasets_hub_dir / "smplx_models"
 smplx_dest.mkdir(parents=True, exist_ok=True)
 
 if not any(smplx_dest.iterdir()):
-    smplx_sources = list(CKPT_DIR.glob("**/smplx_models"))
+    smplx_sources = list(ASSET_CACHE_DIR.glob("**/smplx_models"))
     if smplx_sources:
         smplx_source = smplx_sources[0]
         for item in smplx_source.iterdir():
@@ -57,7 +107,7 @@ if not any(smplx_dest.iterdir()):
                 shutil.copy2(item, target)
 
 if not any(smplx_dest.iterdir()):
-    archives = list(CKPT_DIR.glob("**/smplx_models.*"))
+    archives = list(ASSET_CACHE_DIR.glob("**/smplx_models.*"))
     for archive in archives:
         try:
             if zipfile.is_zipfile(archive):