Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
c69e375
1
Parent(s):
c1a5086
fix
Browse files
app.py
CHANGED
|
@@ -28,6 +28,7 @@ with block:
|
|
| 28 |
point_labels = gr.State(point_labels_empty)
|
| 29 |
masks = gr.State([])
|
| 30 |
cutout_idx = gr.State(set())
|
|
|
|
| 31 |
|
| 32 |
# UI
|
| 33 |
with gr.Column():
|
|
@@ -89,7 +90,7 @@ with block:
|
|
| 89 |
|
| 90 |
def on_everything_image_select(input_image, everything_image, masks, text, evt: gr.SelectData):
|
| 91 |
i = evt.index
|
| 92 |
-
mask =
|
| 93 |
print(mask)
|
| 94 |
print(type(mask))
|
| 95 |
masks.append((mask, text))
|
|
@@ -132,8 +133,8 @@ with block:
|
|
| 132 |
sam_masks = sam.segment_everything(image)
|
| 133 |
print(image)
|
| 134 |
print(sam_masks)
|
| 135 |
-
return
|
| 136 |
-
sam_sgmt_everything_btn.click(on_sam_sgmt_everything_btn_click, components, [everything_image], queue=False)
|
| 137 |
|
| 138 |
|
| 139 |
if __name__ == '__main__':
|
|
|
|
| 28 |
point_labels = gr.State(point_labels_empty)
|
| 29 |
masks = gr.State([])
|
| 30 |
cutout_idx = gr.State(set())
|
| 31 |
+
everything_masks = gr.State([])
|
| 32 |
|
| 33 |
# UI
|
| 34 |
with gr.Column():
|
|
|
|
| 90 |
|
| 91 |
def on_everything_image_select(input_image, everything_image, masks, text, evt: gr.SelectData):
|
| 92 |
i = evt.index
|
| 93 |
+
mask = everything_masks[i][0]
|
| 94 |
print(mask)
|
| 95 |
print(type(mask))
|
| 96 |
masks.append((mask, text))
|
|
|
|
| 133 |
sam_masks = sam.segment_everything(image)
|
| 134 |
print(image)
|
| 135 |
print(sam_masks)
|
| 136 |
+
return [(image, sam_masks), sam_masks]
|
| 137 |
+
sam_sgmt_everything_btn.click(on_sam_sgmt_everything_btn_click, components, [everything_image, everything_masks], queue=False)
|
| 138 |
|
| 139 |
|
| 140 |
if __name__ == '__main__':
|