Add inner repo path to PYTHONPATH

app.py

@@ -34,7 +34,11 @@ def generate_motion(prompt: str) -> str:
     #     'utils.*' imports inside the script can be resolved
     env = os.environ.copy()
     root = Path(__file__).parent
-    env["PYTHONPATH"] = f"{env.get('PYTHONPATH', '')}:{root}"
+    repo_inner = root / "motion_diffusion_model"
+    env["PYTHONPATH"] = (
+        f"{env.get('PYTHONPATH', '')}:{root}:{repo_inner}"
+    )
+
 
     completed = subprocess.run(cmd, env=env, capture_output=True, text=True)
     if completed.returncode != 0: