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__':
|