fix

app.py

@@ -51,6 +51,7 @@ 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)
 
@@ -66,19 +67,14 @@ 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)
-
     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]))
+        generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
         annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
         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)