Update hf_space/app.py

hf_space/app.py

@@ -26,6 +26,12 @@ for entry in sys.path:
     print("  ", entry)
 ASSET_CACHE_DIR = BASE_DIR / "asset_cache"
 ASSET_CACHE_DIR.mkdir(parents=True, exist_ok=True)
+CKPT_DIR = ROOT_DIR / "ckpt"
+CKPT_DIR.mkdir(parents=True, exist_ok=True)
+MEAN_STD_DIR = ROOT_DIR / "mean_std"
+MEAN_STD_DIR.mkdir(parents=True, exist_ok=True)
+WEIGHTS_DIR = ROOT_DIR / "weights"
+WEIGHTS_DIR.mkdir(parents=True, exist_ok=True)
 
 weights_repo = os.environ.get("HF_GESTURELSM_WEIGHTS_REPO", "").strip()
 if weights_repo:
@@ -67,6 +73,25 @@ def _sync_downloaded_assets(download_root: Path) -> None:
 
     anchor_dirs = {"ckpt", "mean_std", "datasets", "weights", "hf_assets"}
 
+    fallback_destinations = {
+        "net_300000_upper.pth": CKPT_DIR / "net_300000_upper.pth",
+        "net_300000_lower.pth": CKPT_DIR / "net_300000_lower.pth",
+        "net_300000_face.pth": CKPT_DIR / "net_300000_face.pth",
+        "net_300000_hands.pth": CKPT_DIR / "net_300000_hands.pth",
+        "net_300000_lower_trans.pth": CKPT_DIR / "net_300000_lower_trans.pth",
+        "face_all_speakers.pth": CKPT_DIR / "face_all_speakers.pth",
+        "hand_all_speakers.pth": CKPT_DIR / "hand_all_speakers.pth",
+        "lower_all_speakers.pth": CKPT_DIR / "lower_all_speakers.pth",
+        "upper_all_speakers.pth": CKPT_DIR / "upper_all_speakers.pth",
+        "meanflow.pth": CKPT_DIR / "meanflow.pth",
+        "new_540_shortcut.bin": CKPT_DIR / "new_540_shortcut.bin",
+        "beatx_2_330_mean.npy": MEAN_STD_DIR / "beatx_2_330_mean.npy",
+        "beatx_2_330_std.npy": MEAN_STD_DIR / "beatx_2_330_std.npy",
+        "beatx_2_trans_mean.npy": MEAN_STD_DIR / "beatx_2_trans_mean.npy",
+        "beatx_2_trans_std.npy": MEAN_STD_DIR / "beatx_2_trans_std.npy",
+        "AESKConv_240_100.bin": WEIGHTS_DIR / "AESKConv_240_100.bin",
+    }
+
     copied = 0
     skipped = 0
     for file_path in download_root.rglob("*"):
@@ -77,10 +102,13 @@ def _sync_downloaded_assets(download_root: Path) -> None:
         relative = file_path.relative_to(download_root)
         parts = relative.parts
         anchor_index = next((idx for idx, part in enumerate(parts) if part in anchor_dirs), None)
-        if anchor_index is None:
-            skipped += 1
-            continue
-        destination = ROOT_DIR / Path(*parts[anchor_index:])
+        if anchor_index is not None:
+            destination = ROOT_DIR / Path(*parts[anchor_index:])
+        else:
+            destination = fallback_destinations.get(file_path.name)
+            if destination is None:
+                skipped += 1
+                continue
         destination.parent.mkdir(parents=True, exist_ok=True)
         try:
             shutil.copy2(file_path, destination)