added cache examples

app.py

@@ -114,6 +114,12 @@ def rotate_btn_fn(img_path, xt, yt, zt, add_bone_cmap=False):
 
 def use_current_btn_fn(input_img):
     return input_img
+
+def retrieve_examples(examples, inputs):
+    global current_img
+    if current_img is not None:
+        return current_img
+    return examples[0]
     
 css_style = "./style.css"
 callback = gr.CSVLogger()
@@ -132,7 +138,7 @@ with gr.Blocks(css=css_style) as app:
                 inputs = [input_img],
                 label = "Xray Examples",
                 elem_id='examples',
-                cache_examples=True,
+                cache_examples="lazy",
                 outputs=output_img
             )
             gr.Examples(
@@ -140,7 +146,7 @@ with gr.Blocks(css=css_style) as app:
                 inputs = [input_img],
                 label = "DRR Examples",
                 elem_id='examples',
-                cache_examples=True,
+                cache_examples="lazy",
                 outputs=output_img
             )
         with gr.Row():