fisk

app.py

@@ -79,9 +79,9 @@ with block:
         image = inputs[raw_image]
         #sam.encode(image)
         generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
-        generated_mask = Image.fromarray(generated_masks)
+        #generated_mask = Image.fromarray(generated_masks)
         #annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
-        return {masks_annotated_image: (image, (generated_mask, "fisk")), 
+        return {masks_annotated_image: (image, [(generated_masks, "fisk")]), 
                 masks: generated_masks, 
                 cutout_idx: set()}
     sam_encode_btn.click(on_click_sam_encode_btn, components, [masks_annotated_image, masks, cutout_idx], queue=True)