Update app.py

app.py

@@ -34,14 +34,16 @@ import sys
 sys.path.append(HOLOPART_CODE_DIR)
 sys.path.append(os.path.join(HOLOPART_CODE_DIR, "scripts"))
 
-"""
+
 EXAMPLES = [
-    ["./holopart/assets/example_data/000.glb", "./holopart/assets/example_data/000.png"],
-    ["./holopart/assets/example_data/001.glb", "./holopart/assets/example_data/001.png"],
-    ["./holopart/assets/example_data/002.glb", "./holopart/assets/example_data/002.png"],
-    ["./holopart/assets/example_data/003.glb", "./holopart/assets/example_data/003.png"],
+    ["./assets/example_data/000.glb", "./assets/example_data/000.PNG"],
+    ["./assets/example_data/001.glb", "./assets/example_data/001.PNG"],
+    ["./assets/example_data/002.glb", "./assets/example_data/002.PNG"],
+    ["./assets/example_data/003.glb", "./assets/example_data/003.PNG"],
+    ["./assets/example_data/004.glb", "./assets/example_data/004.PNG"],
+    ["./assets/example_data/005.glb", "./assets/example_data/005.PNG"]
 ]
-"""
+
 
 
 HEADER = """
@@ -147,7 +149,7 @@ def visualize_input(mesh_in):
 
 
 @spaces.GPU(duration=600)
-def run_example(data_path: str, example_image_path, seed=42, num_inference_steps=25, guidance_scale=3.5):
+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)
@@ -226,16 +228,15 @@ with gr.Blocks(title="HoloPart") as demo:
             model_output = gr.Model3D(label="Decomposed GLB", interactive=False)
             exploded_parts_output = gr.Model3D(label="Exploded Parts", interactive=False)
 
-    """
+   
     with gr.Row():
         examples = gr.Examples(
             examples=EXAMPLES,
-            fn=run_example,
             inputs=[input_mesh, example_image],
             outputs=[model_output, exploded_parts_output],
-            cache_examples=True,
+            cache_examples="lazy",
         )
-    """
+    
 
 
     input_mesh.upload(
@@ -245,6 +246,13 @@ with gr.Blocks(title="HoloPart") as demo:
         queue=False
     )
 
+    input_mesh.change(
+        fn = visualize_input,
+        inputs = [input_mesh],
+        outputs = [input_visualizer],
+        queue=False
+    )
+
     gen_button.click(
         run_full,
         inputs=[