rollback

app.py

@@ -18,7 +18,6 @@ from src.pose_estimation import load_model_from_config, estimate_poses, estimate
 from src.pose_funcs import find_optimal_poses
 from src.utils import spherical_to_cartesian, elu_to_c2w
 
-
 if torch.cuda.is_available():
     _device_ = 'cuda:0'
 else:
@@ -241,15 +240,6 @@ def run_pose_refinement(cam_vis, image1, image2, anchor_polar, explored_sph, ref
     return final_sph, fig
 
 
-def run_example(cam_vis, image1, image2):
-
-    image1, image2 = run_preprocess(image1, image2, True, 0)
-    anchor_polar, explored_sph, fig, _ = run_pose_exploration(cam_vis, image1, image2, 16, 4, 10, 0)
-
-    return image1, image2, anchor_polar, explored_sph, fig
-
-
-
 _HEADER_ = '''
 # Official 🤗 Gradio Demo for [ID-Pose: Sparse-view Camera Pose Estimation By Inverting Diffusion Models](https://github.com/xt4d/id-pose)
 - ID-Pose accepts input images with NO overlapping appearance. 
@@ -277,13 +267,6 @@ def run_demo():
     demo = gr.Blocks(title='ID-Pose: Sparse-view Camera Pose Estimation By Inverting Diffusion Models')
 
     with demo:
-
-        cam_vis = CameraVisualizer([np.eye(4), np.eye(4)], ['Image 1', 'Image 2'], ['red', 'blue'])
-
-        explored_sph = gr.State()
-        anchor_polar = gr.State()
-        refined_sph = gr.State()
-
         gr.Markdown(_HEADER_)
 
         with gr.Row(variant='panel'):
@@ -344,10 +327,8 @@ def run_demo():
                                 ['data/gradio_demo/circo_0.png', 'data/gradio_demo/circo_1.png'],
                             ],
                             inputs=[input_image1, input_image2],
-                            fn=partial(run_example, cam_vis),
-                            outputs=[processed_image1, processed_image2, anchor_polar, explored_sph, vis_output],
                             label='Examples (Captured)',
-                            cache_examples='lazy',
+                            cache_examples=False,
                             examples_per_page=5
                         )
 
@@ -361,10 +342,8 @@ def run_demo():
                                 ['data/gradio_demo/christ_0.png', 'data/gradio_demo/christ_1.png'],
                             ],
                             inputs=[input_image1, input_image2],
-                            fn=partial(run_example, cam_vis),
-                            outputs=[processed_image1, processed_image2, anchor_polar, explored_sph, vis_output],
                             label='Examples (Internet)',
-                            cache_examples='lazy',
+                            cache_examples=False,
                             examples_per_page=5
                         )
                     
@@ -378,13 +357,17 @@ def run_demo():
                                 ['data/gradio_demo/ride_horse_0.png', 'data/gradio_demo/ride_horse_1.png'],
                             ],
                             inputs=[input_image1, input_image2],
-                            fn=partial(run_example, cam_vis),
-                            outputs=[processed_image1, processed_image2, anchor_polar, explored_sph, vis_output],
                             label='Examples (Generated)',
-                            cache_examples='lazy',
+                            cache_examples=False,
                             examples_per_page=5
                         )
 
+        cam_vis = CameraVisualizer([np.eye(4), np.eye(4)], ['Image 1', 'Image 2'], ['red', 'blue'])
+
+        explored_sph = gr.State()
+        anchor_polar = gr.State()
+        refined_sph = gr.State()
+
         run_btn.click(
             fn=run_preprocess,
             inputs=[input_image1, input_image2, preprocess_chk, seed_value],