Update app.py

app.py

@@ -1,3 +1,17 @@
+import sys, subprocess
+
+# 1) Try importing; if it fails, install your wheel
+try:
+    import torchmcubes
+except ImportError:
+    subprocess.check_call([
+        sys.executable, "-m", "pip", "install",
+        "./wheels/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl"
+    ])
+    # now it's installed, so re-import
+    import torchmcubes
+
+# 2) Now safe to import the rest of your app
 import io
 import os
 import shlex
@@ -21,14 +35,6 @@ from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orienta
 # ------------------------------------------------------------
 # 1. Model & utils initialization (runs at startup)
 # ------------------------------------------------------------
-# Install any local wheels (if needed)
-subprocess.run(
-    shlex.split(
-        "pip install wheel/torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl"
-    ),
-    check=False,
-)
-
 # device
 device = "cuda:0" if torch.cuda.is_available() else "cpu"