lazy cache

.gitignore

@@ -1 +1,2 @@
-__pycache__
+__pycache__
+gradio_cached_examples

app.py

@@ -18,6 +18,7 @@ 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:
@@ -240,6 +241,15 @@ 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. 
@@ -267,6 +277,13 @@ 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'):
@@ -327,8 +344,10 @@ 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=False,
+                            cache_examples='lazy',
                             examples_per_page=5
                         )
 
@@ -342,8 +361,10 @@ 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=False,
+                            cache_examples='lazy',
                             examples_per_page=5
                         )
                     
@@ -357,17 +378,13 @@ 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=False,
+                            cache_examples='lazy',
                             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],