update name wheel to wheels

app.py

@@ -5,6 +5,7 @@ import subprocess
 import tempfile
 import zipfile
 from functools import partial
+import sys
 
 from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 from fastapi.responses import StreamingResponse
@@ -18,6 +19,36 @@ from PIL import Image
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
+# ----------------------------------------------------------------
+# 0. Bootstrap install of torchmcubes wheel if needed
+# ----------------------------------------------------------------
+
+# Compute the path to this file and to your wheels folder
+BASE_DIR   = os.path.dirname(os.path.abspath(__file__))
+WHEELS_DIR = os.path.join(BASE_DIR, "wheels")
+WHL_NAME   = "torchmcubes-0.1.0-cp310-cp310-linux_x86_64.whl"
+WHL_PATH   = os.path.join(WHEELS_DIR, WHL_NAME)
+
+def install_torchmcubes():
+    if not os.path.exists(WHL_PATH):
+        raise FileNotFoundError(f"Cannot find wheel at {WHL_PATH!r}")
+    print(f"[bootstrap] Installing torchmcubes from {WHL_PATH!r} …")
+    try:
+        subprocess.check_call([
+            sys.executable, "-m", "pip", "install",
+            "--upgrade", WHL_PATH
+        ])
+    except subprocess.CalledProcessError as e:
+        print(f"[bootstrap][ERROR] pip install failed: {e}")
+        sys.exit(1)
+
+# Try to import, and if missing, install then re-import
+try:
+    import torchmcubes
+except ImportError:
+    install_torchmcubes()
+    import torchmcubes
+
 # ------------------------------------------------------------
 # 1. Model & utils initialization (runs at startup)
 # ------------------------------------------------------------