Fix example logic

app.py

@@ -84,6 +84,7 @@ post_transforms = Compose(
     ]
 )
 
+
 # Define the predict function for the demo
 def predict(input_file, z_axis, model=model, device=device):
     # Load and process data in MONAI format
@@ -115,6 +116,7 @@ def predict(input_file, z_axis, model=model, device=device):
     
     return [t1c, t1, t2, flair], [tc, wt, et] 
 
+
 # Use blocks to set up a more complex demo
 with gr.Blocks() as demo:
 
@@ -128,13 +130,6 @@ with gr.Blocks() as demo:
     
     # Show the button with custom label
     button = gr.Button("Segment Tumor!")
-
-    # Show examples for the user to try
-    gr.Markdown("Try some examples from MONAI's Decathlon Dataset:")
-    examples = gr.Examples(
-        examples=examples,
-        inputs=[gr.File(), gr.Slider()]
-    )
     
     # Show the input image with different MR sequences
     input_image = gr.Gallery(label='input MRI sequences (T1+, T1, T2, FLAIR)')
@@ -146,9 +141,19 @@ with gr.Blocks() as demo:
         inputs=[input_file, z_axis],
         outputs=[input_image, output_segmentation]
     )
+
+    # Have some example for the user to try out
+    examples = gr.Examples(
+        examples=examples,
+        inputs=[input_file, z_axis],
+        outputs=[input_image, output_segmentation],
+        fn=predict,
+        cache_examples=True
+    )
     
     # Show references at the bottom of the demo
     gr.Markdown(references)
 
+
 # Launch the demo 
 demo.launch()