use prompt-only sampler (sample_text2motion.py)

app.py

@@ -13,25 +13,19 @@ DEVICE    = "cpu"                         # free HF Spaces have no GPU
 
 
 def generate_motion(prompt: str) -> str:
-    """
-    Calls the MDM sampling script and returns the generated BVH path.
-    """
-    # create a unique temp directory for this run
-    out_dir = Path("/tmp") / f"mdm_{uuid.uuid4().hex}"
-    out_dir.mkdir(parents=True, exist_ok=True)
+    out_file = Path("/tmp") / f"{uuid.uuid4().hex}.bvh"
 
     cmd = [
         "python",
         "-m",
-        "motion_diffusion_model.sample.generate",
+        "motion_diffusion_model.sample.sample_text2motion",
         "--model_path", str(CKPT_PATH),
-        "--text_prompt", prompt,
-        "--cuda", "-1",                # -1 = cpu
-        "--num_samples", "1",
-        "--output_dir", str(out_dir),
+        "--prompt",     prompt,
+        "--output",     str(out_file),
+        "--device",     "cpu",
     ]
 
-    # add both repo roots to PYTHONPATH
+    # make repo visible to the subprocess
     env = os.environ.copy()
     root = Path(__file__).parent
     repo_inner = root / "motion_diffusion_model"
@@ -41,11 +35,7 @@ def generate_motion(prompt: str) -> str:
     if completed.returncode != 0:
         raise RuntimeError(f"Inference failed:\n{completed.stderr}")
 
-    # find the .bvh the script just wrote
-    bvh_files = list(out_dir.rglob("*.bvh"))
-    if not bvh_files:
-        raise RuntimeError("No BVH file produced.")
-    return str(bvh_files[0])
+    return str(out_file)