update UI

app.py

@@ -156,9 +156,7 @@ def run_pose_exploration_a(image1, image2, seed_value):
         matcher_ckpt_path=_matcher_ckpt_path_
     )
 
-    fig = None
-
-    return elevs, elev_ranges, fig
+    return elevs, elev_ranges, gr.update(value='Preparation Done!')
 
 
 @spaces.GPU
@@ -297,6 +295,8 @@ def run_demo():
                     with gr.Column(min_width=280):
                         processed_image2 = gr.Image(type='numpy', image_mode='RGB', label='Processed Image 2', width=280, interactive=False)
 
+                with gr.Row():
+                    progress_info = gr.Markdown(None, visible=False)
 
                 with gr.Row():
                     preprocess_chk = gr.Checkbox(True, label='Remove background and recenter object')
@@ -381,10 +381,14 @@ def run_demo():
             fn=run_preprocess,
             inputs=[input_image1, input_image2, preprocess_chk, seed_value],
             outputs=[processed_image1, processed_image2],
+        ).success(
+            fn=lambda: gr.update(value='Preparing Estimation...', visible=True),
+            inputs=[],
+            outputs=[progress_info]
         ).success(
             fn=run_pose_exploration_a,
             inputs=[processed_image1, processed_image2, seed_value],
-            outputs=[elevs, elev_ranges, vis_output]
+            outputs=[elevs, elev_ranges, progress_info]
         ).success(
             fn=partial(run_pose_exploration_b, cam_vis),
             inputs=[processed_image1, processed_image2, elevs, elev_ranges, probe_bsz, adj_bsz, adj_iters, seed_value],