Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
d880d7f
1
Parent(s):
df45dfb
fix
Browse files
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()}
|