ax

app.py

@@ -65,10 +65,8 @@ def load_default_pipeline():
 @spaces.GPU()
 def predict_masks(image, points):
     """Predict a single mask from the image based on selected points."""
-    
     if not points:
         return image  # Return the original image if no points are selected
-
     PREDICTOR = SAM2ImagePredictor.from_pretrained(SAM_MODEL, device=DEVICE)
 
     image_np = np.array(image)
@@ -535,6 +533,8 @@ with gr.Blocks(css=css, fill_height=True) as demo:
             )
             with gr.Row():
                 with gr.Column():
+                    image_input = gr.State()
+                    # Input: ImagePrompter for uploaded image
                     upload_image_input = ImagePrompter(show_label=False)
                 with gr.Column():
                     image_output = gr.Image(label="Segmented Image", type="pil", height=400)
@@ -562,6 +562,11 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                     inputs=None,
                     outputs=load_default_message,
                 )
+
+    upload_image_input.change(
+        fn=lambda img: img, inputs=upload_image_input, outputs=image_input
+    )
+    
     target_ratio.change(
         fn=preload_presets,
         inputs=[target_ratio, width_slider, height_slider],