interactive annotated image

app.py

@@ -37,7 +37,7 @@ with block:
                 input_image = gr.Image(label='Input', type='pil', tool=None) # mirror_webcam = False
                 sam_encode_btn = gr.Button('Encode', variant='primary')
                 sam_encode_status = gr.Label('Not encoded yet')
-            masks_annotated_image = gr.AnnotatedImage(label='Segments', height=512)
+            masks_annotated_image = gr.AnnotatedImage(label='Segments')
             with gr.Column():
                 pcl_figure = gr.Model3D(label="3-D Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0])
                 n_samples = gr.Slider(minimum=1e3, maximum=1e6, step=1e3, default=1e3, label='Number of Samples')
@@ -92,7 +92,7 @@ with block:
                  point_labels,
                  image_edit_trigger]
     
-    input_image.select(on_input_image_select,
+    masks_annotated_image.select(on_input_image_select,
                        [raw_image, input_image, image_edit_trigger, point_coords, point_labels, point_label_radio],
                        [raw_image, input_image, point_coords, point_labels, image_edit_trigger], queue=False)