Add nvidia-cuda-toolkit to packages.txt; restore CUDA packages to requirements.txt; clean up app.py

app.py

@@ -1,6 +1,5 @@
 import sys
 import os
-import subprocess
 import tempfile
 import shutil
 import traceback
@@ -8,32 +7,6 @@ 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"
-    )
-
-try:
-    import detectron2  # noqa: F401
-except (ImportError, Exception):
-    _runtime_install(
-        "detectron2 @ git+https://github.com/facebookresearch/detectron2.git@8a9d885b3d4dcf1bef015f0593b872ed8d32b4ab"
-    )
-
 import cv2
 import gradio as gr
 import spaces

packages.txt

@@ -8,3 +8,4 @@ cmake
 ninja-build
 build-essential
 pkg-config
+nvidia-cuda-toolkit

requirements.txt

@@ -8,11 +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 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 requires CUDA at compile time; installed at runtime in app.py
-# detectron2 @ git+https://github.com/facebookresearch/detectron2.git@8a9d885b3d4dcf1bef015f0593b872ed8d32b4ab
+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
 accelerate