fix app to use demo.py and update requirements

app.py

@@ -1,57 +1,43 @@
-import subprocess
-import uuid
-import os
+import subprocess, uuid, os
 from pathlib import Path
 import gradio as gr
 
+# ── paths ─────────────────────────────────────────────────────
+CKPT   = "checkpoints/mld_humanml.pt"       # your 90 MB weight
+DEVICE = "cpu"                              # free HF Spaces
+STEPS  = "50"                               # diffusion steps
+CFG    = "configs/config_mld_humanml3d.yaml"  # default config file
 # ──────────────────────────────────────────────────────────────
-# CONFIG
-# ──────────────────────────────────────────────────────────────
-CKPT   = "checkpoints/mld_humanml.pt"   # <— make sure this file exists
-DEVICE = "cpu"                          # free HF Spaces = CPU only
-STEPS  = "50"                           # diffusion steps (≈60 s on CPU)
-# ──────────────────────────────────────────────────────────────
-
 
 def generate_motion(prompt: str) -> str:
-    """
-    Call the MLD text-to-motion sampler and return a BVH path for Gradio.
-    """
-    # unique temp-file to hold the BVH
     out_bvh = Path("/tmp") / f"{uuid.uuid4().hex}.bvh"
 
     cmd = [
         "python",
-        "-m", "motion_latent_diffusion.mld.sample.text2motion",  # ← sampler module
+        "-m", "motion_latent_diffusion.demo",   # ← new: run demo.py
+        "--cfg",        CFG,
         "--checkpoint", CKPT,
         "--prompt",     prompt,
         "--device",     DEVICE,
-        "--output",     str(out_bvh),
-        "--steps",      STEPS
+        "--steps",      STEPS,
+        "--output",     str(out_bvh)
     ]
 
-    # make sure Python can import motion_latent_diffusion.*
+    # make repo importable
     env = os.environ.copy()
-    repo_root = Path(__file__).parent
-    env["PYTHONPATH"] = f"{env.get('PYTHONPATH', '')}:{repo_root}"
+    root = Path(__file__).parent
+    env["PYTHONPATH"] = f"{env.get('PYTHONPATH','')}:{root}"
 
-    # run the sampler
     subprocess.run(cmd, env=env, check=True)
-
     return str(out_bvh)
 
-
-# ──────────────────  Gradio UI  ───────────────────────────────
+# ── Gradio UI ─────────────────────────────────────────────────
 iface = gr.Interface(
     fn=generate_motion,
-    inputs=gr.Textbox(
-        lines=2,
-        placeholder="e.g. a person walks forward and waves",
-        label="Text Prompt"
-    ),
+    inputs=gr.Textbox(lines=2, placeholder="e.g. a person walks and waves"),
     outputs=gr.File(label="Download BVH"),
-    title="Motion-Latent-Diffusion – Text → 3D Motion",
-    description="Enter a natural-language prompt and get a 3-second BVH animation (CPU demo)."
+    title="Motion-Latent-Diffusion – Text → BVH (CPU demo)",
+    description="Enter a prompt to generate a 3-second motion clip."
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -13,4 +13,5 @@ scipy>=1.10.1
 tensorboard>=2.12.3
 torch>=2.0.0
 tqdm>=4.65.0
-transformers>=4.28.1
+transformers>=4.28.1
+git+https://github.com/openai/CLIP.git