use DiP sampler + add HumanML3D stats

app.py

@@ -13,30 +13,28 @@ DEVICE    = "cpu"                         # free HF Spaces have no GPU
 
 
 def generate_motion(prompt: str) -> str:
-    out_file = Path("/tmp") / f"{uuid.uuid4().hex}.bvh"
+    out_dir = Path("/tmp") / f"mdm_{uuid.uuid4().hex}"
+    out_dir.mkdir(parents=True, exist_ok=True)
 
     cmd = [
-        "python",
-        "-m",
-        "motion_diffusion_model.sample.sample_text2motion",
+        "python", "-m", "motion_diffusion_model.sample.generate",
         "--model_path", str(CKPT_PATH),
-        "--prompt",     prompt,
-        "--output",     str(out_file),
-        "--device",     "cpu",
+        "--text_prompt", prompt,
+        "--cuda", "-1",              # -1  =>  CPU
+        "--guidance_param", "7.5",
+        "--output_dir", str(out_dir),
+        "--num_samples", "1"
     ]
 
-    # make repo visible to the subprocess
     env = os.environ.copy()
     root = Path(__file__).parent
-    repo_inner = root / "motion_diffusion_model"
-    env["PYTHONPATH"] = f"{env.get('PYTHONPATH', '')}:{root}:{repo_inner}"
+    repo = root / "motion_diffusion_model"
+    env["PYTHONPATH"] = f"{env.get('PYTHONPATH','')}:{root}:{repo}"
 
-    completed = subprocess.run(cmd, env=env, capture_output=True, text=True)
-    if completed.returncode != 0:
-        raise RuntimeError(f"Inference failed:\n{completed.stderr}")
-
-    return str(out_file)
+    subprocess.run(cmd, env=env, check=True)
 
+    bvh = next(out_dir.rglob("*.bvh"))
+    return str(bvh)