fix

app.py

@@ -51,7 +51,6 @@ with block:
 
     def on_input_image_upload(input_image):
         # encode image on upload
-        sam.encode(input_image)
         return input_image, point_coords_empty(), point_labels_empty(), None
     input_image.upload(on_input_image_upload, [input_image], [raw_image, point_coords, point_labels], queue=False)
 
@@ -67,6 +66,12 @@ with block:
         return img, point_coords, point_labels
     input_image.select(on_input_image_select, [input_image, point_coords, point_labels, point_label_radio], [input_image, point_coords, point_labels], queue=False)
 
+    # event - inference
+    def on_click_sam_encode_btn(inputs):
+        image = inputs[raw_image]
+        sam.encode(image)
+        return [None, None, None]
+
     def on_click_sam_dencode_btn(inputs):
         image = inputs[raw_image]
         generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
@@ -74,7 +79,7 @@ with block:
         return {masks_annotated_image:annotated, 
                 masks: generated_masks, 
                 cutout_idx: set()}
-    #sam_encode_btn.click(on_click_sam_encode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)
+    sam_encode_btn.click(on_click_sam_encode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)
     sam_decode_btn.click(on_click_sam_dencode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)
     #sam_sgmt_everything_btn.click(on_sam_sgmt_everything_click, components, [masks_annotated_image, masks, cutout_idx], queue=True)