add

app.py

@@ -2,7 +2,35 @@ import os
 # this is a HF Spaces specific hack, as
 #  (i)  building pytorch3d with GPU support is a bit tricky here
 #  (ii) installing the wheel via requirements.txt breaks ZeroGPU
-os.system("pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu118_pyt240/download.html")
+
+# Use the dynamic approach from PyTorch3D documentation to determine the correct wheel
+import sys
+import torch
+
+# Install iopath first as it's a dependency
+os.system("pip install iopath")
+
+# Dynamically determine the correct wheel URL based on Python and PyTorch versions
+pyt_version_str = torch.__version__.split("+")[0].replace(".", "")
+cuda_version_str = torch.version.cuda.replace(".", "") if torch.cuda.is_available() else "cpu"
+version_str = f"py3{sys.version_info.minor}_cu{cuda_version_str}_pyt{pyt_version_str}"
+
+print(f"Looking for PyTorch3D wheel for: {version_str}")
+os.system(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
+
+# If the wheel installation fails, try installing from source
+try:
+    import pytorch3d
+    print("PyTorch3D installed successfully from wheel")
+except ImportError:
+    print("Wheel not found, attempting to install PyTorch3D from source")
+    os.system('pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"')
+    try:
+        import pytorch3d
+        print("PyTorch3D installed successfully from source")
+    except ImportError:
+        print("WARNING: Could not install PyTorch3D. Some functionality may be limited.")
+        # Continue without PyTorch3D - the app will need to handle missing functionality
 
 import torch
 import torch.nn as nn