Deploy Zoo3D with Zero GPU support

app.py

@@ -33,14 +33,14 @@ def _launch():
         os.environ["MAX_IMAGES"] = "10"  # Limit max images for Zero GPU
         os.environ["MAX_RESOLUTION"] = "512"  # Lower resolution for memory optimization
 
-    # Verify critical dependencies are available (installed during build via postBuild)
+    # Verify critical dependencies are available (installed during build via requirements.txt)
     print("🔍 Verifying dependencies...")
     try:
         import detectron2
         print(f"✅ detectron2 available (version: {getattr(detectron2, '__version__', 'unknown')})")
     except ImportError as e:
         print(f"⚠️ detectron2 not available: {e}")
-        print("   detectron2 should be installed via postBuild")
+        print("   detectron2 should be installed via requirements.txt")
 
     try:
         import mmcv
@@ -53,7 +53,7 @@ def _launch():
         print(f"✅ pytorch3d available (version: {getattr(pytorch3d, '__version__', 'unknown')})")
     except ImportError as e:
         print(f"⚠️ pytorch3d not available: {e}")
-        print("   pytorch3d should be installed via postBuild")
+        print("   pytorch3d should be installed via requirements.txt")
 
     # HF Spaces/Gradio sometimes calls /api_info regardless of `show_api=False`.
     # Some gradio_client versions crash when JSON schema uses boolean `additionalProperties`.

packages.txt

@@ -4,3 +4,8 @@ libxext6
 libgl1
 libglib2.0-0
 libgomp1
+cmake
+ninja-build
+g++
+gcc
+git

postBuild

@@ -1,104 +0,0 @@
-#!/usr/bin/env bash
-set -euo pipefail
-
-echo "=== postBuild: Installing build dependencies ==="
-python -m pip install -U pip setuptools wheel ninja cython
-
-# Print environment info
-python - <<'PY' || true
-import torch, platform, sys
-print("Torch:", torch.__version__)
-print("CUDA available:", torch.cuda.is_available())
-print("torch.version.cuda:", getattr(torch.version, "cuda", None))
-print("Python:", sys.version.split()[0])
-print("Platform:", platform.platform())
-PY
-
-# Determine CUDA tag for detectron2 wheel
-PYTORCH_MM="$(python - <<'PY'
-import torch
-v = torch.__version__.split('+')[0].split('.')
-print(f"{v[0]}.{v[1]}")
-PY
-)"
-CUDA_TAG="$(python - <<'PY'
-import torch
-cv = getattr(torch.version, "cuda", None)
-print(("cu"+cv.replace(".","")) if cv else "cpu")
-PY
-)"
-
-echo "=== Installing detectron2 ==="
-WHEEL_URL="https://dl.fbaipublicfiles.com/detectron2/wheels/${CUDA_TAG}/torch${PYTORCH_MM}/index.html"
-echo "Attempting detectron2 wheel from: ${WHEEL_URL}"
-
-if pip install --no-cache-dir "detectron2" -f "${WHEEL_URL}"; then
-  echo "detectron2 installed from prebuilt wheel"
-else
-  echo "Prebuilt wheel not available. Building detectron2 from source..."
-  export FORCE_CUDA=0
-  export MAX_JOBS=${MAX_JOBS:-1}
-  pip install --no-cache-dir --no-build-isolation "git+https://github.com/facebookresearch/detectron2.git@fd27788985af0f4ca800bca563acdb700bb890e2"
-fi
-
-# Verify detectron2 installation
-python - <<'PY'
-import importlib
-spec = importlib.util.find_spec("detectron2")
-if spec is None:
-    raise SystemExit("detectron2 not found after installation")
-import detectron2
-print("✅ detectron2 installed. Version:", getattr(detectron2, "__version__", "unknown"))
-PY
-
-echo "=== Installing pytorch3d ==="
-# pytorch3d requires special handling - try prebuilt wheels first
-PYTORCH3D_WHEEL_URL="https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_${CUDA_TAG}_pyt${PYTORCH_MM}/download.html"
-echo "Attempting pytorch3d wheel from: ${PYTORCH3D_WHEEL_URL}"
-
-# Try prebuilt wheel first, fallback to source build
-if pip install --no-cache-dir "pytorch3d" -f "${PYTORCH3D_WHEEL_URL}" 2>/dev/null; then
-  echo "pytorch3d installed from prebuilt wheel"
-else
-  echo "Prebuilt pytorch3d wheel not available. Building from source..."
-  export FORCE_CUDA=${FORCE_CUDA:-0}
-  export MAX_JOBS=${MAX_JOBS:-1}
-  pip install --no-cache-dir --no-build-isolation "git+https://github.com/facebookresearch/pytorch3d.git"
-fi
-
-# Verify pytorch3d installation
-python - <<'PY'
-try:
-    import pytorch3d
-    print("✅ pytorch3d installed. Version:", getattr(pytorch3d, "__version__", "unknown"))
-except ImportError as e:
-    print(f"⚠️ pytorch3d import failed (may work at runtime): {e}")
-PY
-
-echo "=== Installing vendored detectron2 (CropFormer) ==="
-# Install vendored detectron2 with CropFormer extensions
-if [ -d "MaskClustering/third_party/detectron2" ]; then
-  export MAX_JOBS=${MAX_JOBS:-1}
-  pip install --no-build-isolation -e MaskClustering/third_party/detectron2 || echo "⚠️ Vendored detectron2 installation failed (will use system detectron2)"
-fi
-
-echo "=== Compiling CropFormer components ==="
-CROPFORMER_ROOT="MaskClustering/third_party/detectron2/projects/CropFormer"
-
-# 1. Compile entity_api PythonAPI
-if [ -d "${CROPFORMER_ROOT}/entity_api/PythonAPI" ]; then
-  echo "Compiling entity_api PythonAPI..."
-  (cd "${CROPFORMER_ROOT}/entity_api/PythonAPI" && make) || echo "⚠️ entity_api compilation failed (non-critical)"
-fi
-
-# 2. Compile MSDeformAttn CUDA ops
-OPS_DIR="${CROPFORMER_ROOT}/mask2former/modeling/pixel_decoder/ops"
-if [ -d "${OPS_DIR}" ]; then
-  echo "Compiling MSDeformAttn CUDA operators..."
-  (cd "${OPS_DIR}" && sh make.sh) || {
-    echo "⚠️ CUDA compilation failed, trying CPU-only mode..."
-    (cd "${OPS_DIR}" && FORCE_CUDA=0 python setup.py build install) || echo "⚠️ MSDeformAttn compilation failed (non-critical)"
-  }
-fi
-
-echo "=== postBuild completed ==="

requirements.txt

@@ -28,10 +28,22 @@ timm
 h5py
 scikit-image
 
+# pytorch3d dependencies (required before installing pytorch3d)
+fvcore>=0.1.5,<0.1.6
+iopath>=0.1.7,<0.1.10
 
-# detectron2 runtime deps (needed when importing vendored detectron2 via PYTHONPATH)
+# detectron2 dependencies (install before detectron2)
+pycocotools>=2.0.2
+termcolor>=1.1
+yacs>=0.1.8
+tabulate
+cloudpickle
+tensorboard
+black
+packaging
 
+# pytorch3d - install from git (will be compiled during build)
+pytorch3d @ git+https://github.com/facebookresearch/pytorch3d.git@stable
 
-# NOTE (HF Spaces): do not use local-path installs here.
-# HF builds install requirements BEFORE copying the repo into /app.
-# detectron2 is installed from vendored source in `postBuild`.
+# detectron2 - install from local third_party directory
+./MaskClustering/third_party/detectron2