Update hf_space/app.py

hf_space/app.py

@@ -1,4 +1,5 @@
 import os
+import shutil
 import sys
 from pathlib import Path
 
@@ -35,9 +36,26 @@ if weights_repo:
     )
 
 # Ensure expected runtime directories exist so the demo can write outputs.
-for relative in ["outputs/audio2pose", "datasets/BEAT_SMPL"]:
+datasets_hub_dir = BASE_DIR / "datasets" / "hub"
+for relative in ["outputs/audio2pose", "datasets/BEAT_SMPL", "datasets/hub"]:
     (BASE_DIR / relative).mkdir(parents=True, exist_ok=True)
 
+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"))
+    if smplx_sources:
+        smplx_source = smplx_sources[0]
+        for item in smplx_source.iterdir():
+            target = smplx_dest / item.name
+            if item.is_dir():
+                shutil.copytree(item, target, dirs_exist_ok=True)
+            else:
+                shutil.copy2(item, target)
+    else:
+        print("[GestureLSM] WARNING: smplx_models directory missing; ensure the weights repo contains it.")
+
 # Reuse the existing Gradio interface defined in demo.py.
 from demo import demo as gesture_demo  # noqa: E402