Update

app.py

@@ -29,6 +29,7 @@ from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import postprocessing_utils, render_utils
 
 MAX_SEED = np.iinfo(np.int32).max
+TEMP_DIR = gr.utils.get_upload_folder()
 
 pipeline = TrellisImageTo3DPipeline.from_pretrained("microsoft/TRELLIS-image-large")
 pipeline.cuda()
@@ -177,8 +178,8 @@ def image_to_3d(
     video = [np.concatenate([video[i], video_geo[i]], axis=1) for i in range(len(video))]
 
     with (
-        tempfile.NamedTemporaryFile(suffix=".pth", delete=False) as state_file,
-        tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as video_file,
+        tempfile.NamedTemporaryFile(suffix=".pth", dir=TEMP_DIR, delete=False) as state_file,
+        tempfile.NamedTemporaryFile(suffix=".mp4", dir=TEMP_DIR, delete=False) as video_file,
     ):
         save_state_to_file(outputs["gaussian"][0], outputs["mesh"][0], state_file.name)
         torch.cuda.empty_cache()
@@ -205,7 +206,7 @@ def extract_glb(
     gs, mesh = load_state_from_file(state_path)
     glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
     torch.cuda.empty_cache()
-    with tempfile.NamedTemporaryFile(suffix=".glb", delete=False) as glb_file:
+    with tempfile.NamedTemporaryFile(suffix=".glb", dir=TEMP_DIR, delete=False) as glb_file:
         glb.export(glb_file.name)
         return glb_file.name
 
@@ -221,7 +222,7 @@ def extract_gaussian(state_path: str) -> str:
         str: The path to the extracted Gaussian file.
     """
     gs, _ = load_state_from_file(state_path)
-    with tempfile.NamedTemporaryFile(suffix=".ply", delete=False) as gaussian_file:
+    with tempfile.NamedTemporaryFile(suffix=".ply", dir=TEMP_DIR, delete=False) as gaussian_file:
         gs.save_ply(gaussian_file.name)
         return gaussian_file.name
 
@@ -278,7 +279,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             model_output = gr.Model3D(label="Extracted GLB/Gaussian", height=300)
 
-    state_file_path = gr.Textbox(visible=False)
+    state_file = gr.File(visible=False)
 
     examples = gr.Examples(
         examples=sorted(pathlib.Path("assets/example_image").glob("*.png")),
@@ -309,7 +310,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             slat_guidance_strength,
             slat_sampling_steps,
         ],
-        outputs=[state_file_path, video_output],
+        outputs=[state_file, video_output],
     ).then(
         fn=lambda: (gr.Button(interactive=True), gr.Button(interactive=True)),
         outputs=[extract_glb_btn, extract_gs_btn],
@@ -322,8 +323,8 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         api_name=False,
     )
 
-    extract_glb_btn.click(fn=extract_glb, inputs=[state_file_path, mesh_simplify, texture_size], outputs=model_output)
-    extract_gs_btn.click(fn=extract_gaussian, inputs=state_file_path, outputs=model_output)
+    extract_glb_btn.click(fn=extract_glb, inputs=[state_file, mesh_simplify, texture_size], outputs=model_output)
+    extract_gs_btn.click(fn=extract_gaussian, inputs=state_file, outputs=model_output)
 
 if __name__ == "__main__":
     demo.launch(mcp_server=True)