simplify

app.py

@@ -115,7 +115,7 @@ def explode_mesh(mesh: trimesh.Scene, explode_factor: float = 0.5):
 
 
 @spaces.GPU(duration=600)
-def run_full(data_path, seed=42, num_inference_steps=25, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
+def run_full(data_path: str, example_image=None, seed=42, num_inference_steps=25, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
 
     batch_size = 30
     parts_data = prepare_data(data_path)
@@ -148,38 +148,6 @@ def visualize_input(mesh_in):
     return mesh_in
 
 
-@spaces.GPU(duration=600)
-def run_example(data_path: str, example_image, seed=42, num_inference_steps=25, guidance_scale=3.5):
-
-    batch_size = 30
-    parts_data = prepare_data(data_path)
-
-    part_scene = run_holopart(
-        holopart_pipe,
-        batch=parts_data,
-        batch_size=batch_size,
-        seed=seed,
-        num_inference_steps=num_inference_steps,
-        guidance_scale=guidance_scale,
-        num_chunks=1000000,
-    )
-    print("mesh extraction done")
-
-    
-    save_dir = os.path.join(TMP_DIR, "examples")
-    os.makedirs(save_dir, exist_ok=True)
-    mesh_path = os.path.join(save_dir, f"holorpart_{get_random_hex()}.glb")
-    part_scene.export(mesh_path)
-    print("save to ", mesh_path)
-    exploded_mesh = explode_mesh(part_scene, 0.5)
-    exploded_mesh_path = os.path.join(save_dir, f"holorpart_exploded_{get_random_hex()}.glb")
-    exploded_mesh.export(exploded_mesh_path)
-
-    torch.cuda.empty_cache()
-
-    return mesh_path, exploded_mesh_path
-
-
 with gr.Blocks(title="HoloPart") as demo:
     gr.Markdown(HEADER)
 
@@ -232,10 +200,11 @@ with gr.Blocks(title="HoloPart") as demo:
     with gr.Row():
         examples = gr.Examples(
             examples=EXAMPLES,
-            fn=run_example,
+            fn=run_full,
             inputs=[input_mesh, example_image],
             outputs=[model_output, exploded_parts_output],
-            cache_examples="lazy",
+            cache_examples=True,
+            cache_mode="lazy"
         )