Upload 2 files

app.py

@@ -107,7 +107,7 @@ def explode_mesh(mesh: trimesh.Scene, explode_factor: float = 0.5):
     return exploded_mesh
 
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=300)
 def run_full(data_path, seed=42, num_inference_steps=25, guidance_scale=3.5):
 
     batch_size = 30
@@ -138,7 +138,7 @@ def run_full(data_path, seed=42, num_inference_steps=25, guidance_scale=3.5):
     return mesh_path, exploded_mesh_path
 
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=300)
 def run_example(data_path: str, example_image_path, seed=42, num_inference_steps=25, guidance_scale=3.5):
 
     batch_size = 30
@@ -223,6 +223,7 @@ with gr.Blocks(title="HoloPart") as demo:
             inputs=[input_mesh, example_image],
             outputs=[model_output, exploded_parts_output],
             cache_examples=True,
+            cache_mode="lazy",
         )
 
 
@@ -240,4 +241,4 @@ with gr.Blocks(title="HoloPart") as demo:
     demo.load(start_session)
     demo.unload(end_session)
 
-demo.launch()
+demo.launch(ssr_mode=False, allowed_paths=["./tmp/"])

requirements.txt

@@ -1,3 +1,4 @@
+torch==2.4.0
 torchvision
 diffusers
 transformers==4.49.0
@@ -21,5 +22,5 @@ https://huggingface.co/spaces/JeffreyXiang/TRELLIS/resolve/main/wheels/nvdiffras
 cvcuda_cu12
 gltflib
 https://huggingface.co/spaces/VAST-AI/TripoSG/resolve/main/diso-0.1.4-cp310-cp310-linux_x86_64.whl?download=true
---find-links https://data.pyg.org/whl/torch-2.6.0+cu124.html
+--find-links https://data.pyg.org/whl/torch-2.4.0+cu124.html
 torch-cluster