reducing dependancy installations

README.md

@@ -10,16 +10,18 @@ app_file: app.py
 pinned: false
 license: cc-by-nc-4.0
 short_description: Forearm-Guided Camera-Space 3D Hand Pose
-suggested_hardware: a10g-small
 startup_duration_timeout: 1h
 ---
 
-This Space runs the EgoForce video demo with the Hugging Face Gradio SDK.
+This Space runs the EgoForce video demo with the Hugging Face Gradio SDK and is prepared for ZeroGPU hardware.
 
 At startup, `app.py`:
 
 - Clones `https://github.com/dfki-av/EgoForce` into the Space runtime.
+- Patches the upstream Gradio callback with `@spaces.GPU` so video inference runs inside ZeroGPU's dynamic CUDA allocation.
 - Downloads the `_DATA` model assets from `chris10/EgoForce` on the Hugging Face Hub.
 - Launches the upstream Gradio demo at `demo/run_app.py`.
 
 Most Python dependencies are declared in `requirements.txt`, with build prerequisites in `pre-requirements.txt`. Packages that need custom pip flags or repo-local subdirectories are installed by `app.py` after the EgoForce source checkout. Debian packages needed for OpenCV, ffmpeg, headless rendering, and compiled extensions are declared in `packages.txt`.
+
+Select `ZeroGPU` in the Space hardware settings. `ZEROGPU_DURATION_SECONDS` defaults to `600`, and `ZEROGPU_SIZE` defaults to `large`.

app.py

@@ -15,6 +15,8 @@ EGOFORCE_REPO_URL = os.environ.get("EGOFORCE_REPO_URL", "https://github.com/dfki
 EGOFORCE_REF = os.environ.get("EGOFORCE_REF", "main")
 EGOFORCE_ROOT = Path(os.environ.get("EGOFORCE_ROOT", SPACE_ROOT / "EgoForce")).resolve()
 EGOFORCE_ASSETS_REPO_ID = os.environ.get("EGOFORCE_ASSETS_REPO_ID", "chris10/EgoForce")
+ZEROGPU_DURATION_SECONDS = os.environ.get("ZEROGPU_DURATION_SECONDS", "600")
+ZEROGPU_SIZE = os.environ.get("ZEROGPU_SIZE", "large")
 
 
 def run_command(command: list[str], cwd: Path | None = None) -> None:
@@ -26,12 +28,37 @@ def configure_runtime_environment() -> None:
     os.environ.setdefault("EGOFORCE_ROOT", str(EGOFORCE_ROOT))
     configure_cuda_environment()
     os.environ.setdefault("FORCE_CUDA", "1")
-    os.environ.setdefault("TORCH_CUDA_ARCH_LIST", "7.5;8.0;8.6;8.9")
+    configure_torch_cuda_arch_list()
+    os.environ.setdefault("MAX_JOBS", "1")
+    os.environ.setdefault("CMAKE_BUILD_PARALLEL_LEVEL", "1")
+    os.environ.setdefault("NINJAFLAGS", "-j1")
+    os.environ.setdefault("MAKEFLAGS", "-j1")
     os.environ.setdefault("PYOPENGL_PLATFORM", "egl")
     os.environ.setdefault("MPLBACKEND", "Agg")
     os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
 
 
+def configure_torch_cuda_arch_list() -> None:
+    if os.environ.get("TORCH_CUDA_ARCH_LIST"):
+        return
+
+    accelerator = os.environ.get("ACCELERATOR", "").lower()
+    if "zero" in accelerator or "h200" in accelerator:
+        arch_list = "9.0"
+    elif "t4" in accelerator:
+        arch_list = "7.5"
+    elif "a100" in accelerator:
+        arch_list = "8.0"
+    elif "a10g" in accelerator:
+        arch_list = "8.6"
+    elif "l4" in accelerator or "l40" in accelerator:
+        arch_list = "8.9"
+    else:
+        arch_list = "9.0"
+
+    os.environ["TORCH_CUDA_ARCH_LIST"] = arch_list
+
+
 def candidate_site_packages() -> list[Path]:
     paths = [Path(path) for path in site.getsitepackages()]
     user_site = site.getusersitepackages()
@@ -64,6 +91,7 @@ def configure_cuda_environment() -> None:
 def ensure_egoforce_repo() -> Path:
     demo_entrypoint = EGOFORCE_ROOT / "demo" / "run_app.py"
     if demo_entrypoint.exists():
+        patch_upstream_gradio_for_zerogpu(demo_entrypoint)
         return EGOFORCE_ROOT
 
     if EGOFORCE_ROOT.exists() and any(EGOFORCE_ROOT.iterdir()):
@@ -85,9 +113,36 @@ def ensure_egoforce_repo() -> Path:
     if not demo_entrypoint.exists():
         raise RuntimeError(f"EgoForce demo entrypoint not found at {demo_entrypoint}")
 
+    patch_upstream_gradio_for_zerogpu(demo_entrypoint)
     return EGOFORCE_ROOT
 
 
+def patch_upstream_gradio_for_zerogpu(demo_entrypoint: Path) -> None:
+    source = demo_entrypoint.read_text(encoding="utf-8")
+
+    if "import spaces\n" not in source:
+        if "import torch\n" not in source:
+            raise RuntimeError(f"Could not insert ZeroGPU import in {demo_entrypoint}")
+        source = source.replace("import torch\n", "import spaces\nimport torch\n", 1)
+
+    if "@spaces.GPU(" not in source:
+        if "def process_video(\n" not in source:
+            raise RuntimeError(f"Could not locate process_video in {demo_entrypoint}")
+        source = source.replace(
+            "def process_video(\n",
+            (
+                "@spaces.GPU("
+                f"duration={int(ZEROGPU_DURATION_SECONDS)}, "
+                f"size={ZEROGPU_SIZE!r}"
+                ")\n"
+                "def process_video(\n"
+            ),
+            1,
+        )
+
+    demo_entrypoint.write_text(source, encoding="utf-8")
+
+
 def package_available(module_name: str) -> bool:
     return importlib.util.find_spec(module_name) is not None
 
@@ -99,6 +154,7 @@ def pip_install(requirement: str, *extra_args: str) -> None:
         "pip",
         "install",
         "--no-cache-dir",
+        "--disable-pip-version-check",
         requirement,
         *extra_args,
     ]
@@ -108,7 +164,7 @@ def pip_install(requirement: str, *extra_args: str) -> None:
 def ensure_runtime_python_packages(repo_root: Path) -> None:
     datapipes_path = repo_root / "thirdparty" / "datapipes"
     install_plan = [
-        ("mmcv", "mmcv==2.1.0", ("--no-build-isolation",)),
+        ("mmcv", "mmcv==2.1.0", ("--no-build-isolation", "--no-deps")),
         ("anycalib", "git+https://github.com/javrtg/AnyCalib.git", ("--no-build-isolation",)),
         ("chumpy", "git+https://github.com/mattloper/chumpy.git", ("--no-build-isolation",)),
         ("pytorch3d", "git+https://github.com/facebookresearch/pytorch3d.git", ("--no-build-isolation",)),

requirements.txt

@@ -22,6 +22,8 @@ pycocotools==2.0.10
 trimesh==4.11.3
 sortedcontainers==2.4.0
 openmim==0.3.9
+mmengine==0.10.7
+yapf==0.43.0
 lmdb==2.0.0
 ultralytics==8.4.23
 lap==0.5.13
@@ -32,3 +34,4 @@ yacs==0.1.8
 projectaria_client_sdk==1.1.0
 huggingface_hub
 hf_xet
+spaces