Update app.py

app.py

@@ -228,6 +228,10 @@ def initialize_models(sam_path, adapter_pth, model_cfg, cfg):
         mask_adapter.load_state_dict(adapter_state_dict)
         print("Mask Adapter model initialized.")
 
+def preprocess_example(input_img, img_state):
+    img_state.clear()
+    return img_state,None
+
 def clear_everything(img_state):
     img_state.clear()
     return img_state, None, None
@@ -346,9 +350,12 @@ with gr.Blocks() as demo:
             
             gr.Examples(
                     examples=examples_point, 
-                    inputs=input_image, 
-                    outputs=output_image_box, 
+                    inputs=[input_image, img_state_bbox], 
+                    outputs=[img_state_bbox, output_image_box], 
                     examples_per_page=5,
+                    fn=preprocess_example,
+                    run_on_click=True,
+                    cache_examples=False,
                 )
                 
         with gr.TabItem("Point Mode"):
@@ -393,9 +400,16 @@ with gr.Blocks() as demo:
             def clear_and_set_example_point(example):
                 clear_everything(img_state_points)
                 return example
-            with gr.Row():
-                gr.Examples(examples=examples_point, inputs=input_image, outputs=output_image_point,examples_per_page=5, preprocess=clear_and_set_example_point)
-            
+
+            gr.Examples(
+                    examples=examples_point, 
+                    inputs=[input_image, img_state_points], 
+                    outputs=[img_state_points, output_image_point], 
+                    examples_per_page=5,
+                    fn=preprocess_example,
+                    run_on_click=True,
+                    cache_examples=False,
+                )