add tqdm progress + mesh input visualizer

app.py

@@ -113,7 +113,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):
+def run_full(data_path, 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)
@@ -142,6 +142,9 @@ def run_full(data_path, seed=42, num_inference_steps=25, guidance_scale=3.5):
 
     return mesh_path, exploded_mesh_path
 
+def visualize_input(mesh_in):
+    return 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):
@@ -233,6 +236,12 @@ with gr.Blocks(title="HoloPart") as demo:
     """
 
 
+    input_mesh.upload(
+        fn = visualize_input,
+        inputs = [input_mesh],
+        outputs = [input_visualizer],
+        queue=False
+    )
 
     gen_button.click(
         run_full,