jens commited on
Commit
c69e375
·
1 Parent(s): c1a5086
Files changed (1) hide show
  1. app.py +4 -3
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 = everything_image[1][i][0]
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 {everything_image: (image, sam_masks)}
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__':