update

app.py

@@ -23,6 +23,17 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
 # ------------------------------------------------------------
 # Install any local wheels (if needed)
 
+import sys
+
+# Install any local wheel files in the "wheels" directory
+wheels_dir = os.path.join(os.path.dirname(__file__), "wheels")
+if os.path.isdir(wheels_dir):
+    for file in os.listdir(wheels_dir):
+        if file.endswith(".whl"):
+            whl_path = os.path.join(wheels_dir, file)
+            subprocess.check_call([sys.executable, "-m", "pip", "install", whl_path])
+
+
 
 # device
 device = "cuda:0" if torch.cuda.is_available() else "cpu"

requirements.txt

@@ -1,6 +1,3 @@
---find-links ./wheel
-torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl
-
 omegaconf
 Pillow
 einops