use DiP sampler + add HumanML3D stats

app.py

@@ -11,11 +11,10 @@ CKPT_PATH = "checkpoints/t2m_50step.pt"   # make sure this file exists
 DEVICE    = "cpu"                         # free HF Spaces have no GPU
 # ---------------------------------------------------------------------
 
-
 def generate_motion(prompt: str) -> str:
     out_dir = Path("/tmp") / f"mdm_{uuid.uuid4().hex}"
     out_dir.mkdir(parents=True, exist_ok=True)
-
+    
     cmd = [
         "python", "-m", "motion_diffusion_model.sample.generate",
         "--model_path", str(CKPT_PATH),
@@ -24,21 +23,23 @@ def generate_motion(prompt: str) -> str:
         "--guidance_param", "7.5",
         "--output_dir", str(out_dir),
         "--num_samples", "1",
-        "--unconstrained"          # ← add this line
+        "--unconstrained",
+        "--inference_only"  # Add this flag (you'll need to implement it)
     ]
-
-
+    
     env = os.environ.copy()
     root = Path(__file__).parent
     repo = root / "motion_diffusion_model"
     env["PYTHONPATH"] = f"{env.get('PYTHONPATH','')}:{root}:{repo}"
-
-    subprocess.run(cmd, env=env, check=True)
-
-    bvh = next(out_dir.rglob("*.bvh"))
-    return str(bvh)
-
-
+    
+    try:
+        subprocess.run(cmd, env=env, check=True)
+        bvh = next(out_dir.rglob("*.bvh"))
+        return str(bvh)
+    except subprocess.CalledProcessError as e:
+        return f"Error generating motion: {str(e)}"
+    except StopIteration:
+        return "No BVH file was generated."
 
 # -----------------------  Gradio UI  ----------------------------------
 iface = gr.Interface(
@@ -56,4 +57,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()