PIL image

app.py

@@ -79,6 +79,7 @@ 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_masks = Image.fromarray(generated_masks)
         annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
         return {masks_annotated_image: generated_masks, 
                 masks: generated_masks,