jens commited on
Commit
d880d7f
·
1 Parent(s): df45dfb
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ with block:
80
  #sam.encode(image)
81
  generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
82
  generated_masks = Image.fromarray(generated_masks)
83
- annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
84
  return {masks_annotated_image: generated_masks,
85
  masks: generated_masks,
86
  cutout_idx: set()}
 
80
  #sam.encode(image)
81
  generated_masks, _, _ = sam.cond_pred(pts=np.array(inputs[point_coords]), lbls=np.array(inputs[point_labels]))
82
  generated_masks = Image.fromarray(generated_masks)
83
+ #annotated = (image, [(generated_masks[i], f'Mask {i}') for i in range(len(generated_masks))])
84
  return {masks_annotated_image: generated_masks,
85
  masks: generated_masks,
86
  cutout_idx: set()}