update

app.py

@@ -13,7 +13,7 @@ import subprocess
 import shutil
 
 # install others
-subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
+#subprocess.run("pip install spandrel==0.4.1 --no-deps", shell=True, check=True)
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16
@@ -108,7 +108,7 @@ remove_bg_fn = lambda x: remove_bg(x, birefnet, transform_image, DEVICE)
 if not os.path.exists("checkpoints/RealESRGAN_x2plus.pth"):
     hf_hub_download("dtarnow/UPscaler", filename="RealESRGAN_x2plus.pth", local_dir="checkpoints")
 if not os.path.exists("checkpoints/big-lama.pt"):
-    subprocess.run("wget -P checkpoints/ https://github.com/Sanster/models/releases/download/add_big_lama/big-lama.pt", shell=True, check=True)
+    hf_hub_download("cocktailpeanut/biglama", filename="big-lama.pt", local_dir="checkpoints")
 
 def start_session(req: gr.Request):
     save_dir = os.path.join(TMP_DIR, str(req.session_hash))
@@ -465,4 +465,4 @@ with gr.Blocks(title="TripoSG") as demo:
     demo.load(start_session)
     demo.unload(end_session)
 
-demo.launch()
+demo.launch()

requirements.txt

@@ -1,4 +1,6 @@
-torchvision
+gradio
+spaces
+#torchvision
 diffusers
 transformers==4.49.0
 einops
@@ -17,7 +19,8 @@ open3d
 timm
 kornia
 ninja
-https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
-cvcuda_cu12
+#https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffrast-0.3.3-cp310-cp310-linux_x86_64.whl?download=true
+#cvcuda_cu12
 gltflib
-torch-cluster
+#torch-cluster
+spandrel==0.4.1