update torchmcubes

app.py

@@ -1,55 +1,20 @@
+import io
 import os
-import sys
+import shlex
 import subprocess
+import tempfile
+import zipfile
+from functools import partial
 
-# ----------------------------------------------------------------
-# 0. Bootstrap install of torchmcubes wheel if needed
-# ----------------------------------------------------------------
-
-# 1) Locate the wheel relative to this file
-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)
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException
+from fastapi.responses import StreamingResponse
+from fastapi.middleware.cors import CORSMiddleware
 
-def install_torchmcubes_wheel(whl_path: str):
-    """
-    Installs the given wheel via pip. Exits on failure.
-    """
-    if not os.path.exists(whl_path):
-        raise FileNotFoundError(f"Cannot find torchmcubes 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)
-
-# 2) Try import, install if missing
-try:
-    import torchmcubes
-except ImportError:
-    install_torchmcubes_wheel(WHL_PATH)
-    import torchmcubes
-
-# ----------------------------------------------------------------
-# 1. Now safe to import the rest of your dependencies
-# ----------------------------------------------------------------
-import io
-import zipfile
-import tempfile
 import numpy as np
 import rembg
 import torch
 from PIL import Image
 
-from fastapi import FastAPI, File, UploadFile, Form, HTTPException
-from fastapi.responses import StreamingResponse
-from fastapi.middleware.cors import CORSMiddleware
-
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 

requirements.txt

@@ -10,3 +10,5 @@ gradio
 onnxruntime
 fastapi
 uvicorn[standard]
+--no-binary=torchmcubes
+torchmcubes @ git+https://github.com/tatsy/torchmcubes.git