Move nvdiffrast+diso to runtime install (need nvcc, not available at build time)

app.py

@@ -1,5 +1,6 @@
 import sys
 import os
+import subprocess
 import tempfile
 import shutil
 import traceback
@@ -7,6 +8,25 @@ import json
 import random
 from pathlib import Path
 
+# nvdiffrast and diso require CUDA at compile time — install at runtime
+# (ZeroGPU build containers don't have nvcc; runtime containers do)
+def _runtime_install(*packages):
+    subprocess.run([sys.executable, "-m", "pip", "install", "--quiet", *packages], check=True)
+
+try:
+    import nvdiffrast.torch as dr  # noqa: F401
+except (ImportError, Exception):
+    _runtime_install(
+        "nvdiffrast @ git+https://github.com/NVlabs/nvdiffrast.git@253ac4fcea7de5f396371124af597e6cc957bfae"
+    )
+
+try:
+    import diso  # noqa: F401
+except (ImportError, Exception):
+    _runtime_install(
+        "diso @ git+https://github.com/SarahWeiii/diso.git@9792ad928ccb09bdec938779651ee03e395758a6"
+    )
+
 import cv2
 import gradio as gr
 import spaces

requirements.txt

@@ -8,8 +8,9 @@ clip @ git+https://github.com/openai/CLIP.git@d05afc436d78f1c48dc0dbf8e5980a9d47
 mvadapter @ git+https://github.com/huanngzh/MV-Adapter.git@4277e0018232bac82bb2c103caf0893cedb711be
 chumpy @ https://huggingface.co/datasets/Daankular/wheels/resolve/main/chumpy-0.71-py3-none-any.whl
 skel @ git+https://github.com/MarilynKeller/SKEL.git@c32cf16581295bff19399379efe5b776d707cd95
-nvdiffrast @ git+https://github.com/NVlabs/nvdiffrast.git@253ac4fcea7de5f396371124af597e6cc957bfae
-diso @ git+https://github.com/SarahWeiii/diso.git@9792ad928ccb09bdec938779651ee03e395758a6
+# nvdiffrast and diso require CUDA (nvcc) at compile time; installed at runtime in app.py
+# nvdiffrast @ git+https://github.com/NVlabs/nvdiffrast.git@253ac4fcea7de5f396371124af597e6cc957bfae
+# diso @ git+https://github.com/SarahWeiii/diso.git@9792ad928ccb09bdec938779651ee03e395758a6
 detectron2 @ git+https://github.com/facebookresearch/detectron2.git@8a9d885b3d4dcf1bef015f0593b872ed8d32b4ab
 
 # Core ML