Initial MLD text→motion Space

app.py

@@ -0,0 +1,58 @@
+import subprocess
+import uuid
+import os
+from pathlib import Path
+import gradio as gr
+
+# ──────────────────────────────────────────────────────────────
+# 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
+        "--checkpoint", CKPT,
+        "--prompt",     prompt,
+        "--device",     DEVICE,
+        "--output",     str(out_bvh),
+        "--steps",      STEPS
+    ]
+
+    # make sure Python can import motion_latent_diffusion.*
+    env = os.environ.copy()
+    repo_root = Path(__file__).parent
+    env["PYTHONPATH"] = f"{env.get('PYTHONPATH', '')}:{repo_root}"
+
+    # run the sampler
+    subprocess.run(cmd, env=env, check=True)
+
+    return str(out_bvh)
+
+
+# ──────────────────  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"
+    ),
+    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)."
+)
+
+if __name__ == "__main__":
+    iface.launch()

checkpoints/mld_humanml.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cdb21d78905fcaa41212f85b3fe13fa342f84ccbad5f90b9a7251b217134ecc4
+size 1652

motion_latent_diffusion

@@ -0,0 +1 @@
+Subproject commit 081ce3152e5a95b14a4495fb2d32939c257c6216

requirements.txt

@@ -0,0 +1,16 @@
+accelerate>=0.16.0
+configargparse>=1.5.3
+einops>=0.6.0
+gradio>=3.32.0
+h5py>=3.8.0
+huggingface-hub>=0.14.1
+hydra-core>=1.3.2
+matplotlib>=3.7.1
+numpy>=1.24.3
+omegaconf>=2.3.0
+safetensors>=0.3.1
+scipy>=1.10.1
+tensorboard>=2.12.3
+torch>=2.0.0
+tqdm>=4.65.0
+transformers>=4.28.1