jens commited on
Commit
8728327
·
1 Parent(s): ab17802
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -91,6 +91,7 @@ with block:
91
  reset_btn.click(on_reset_btn_click, [input_image], [input_image, point_coords, point_labels], queue=False)
92
 
93
  def on_prompt_image_select(input_image, prompt_image, point_coords, point_labels, point_label_radio, text, pred_masks, embedding, evt: gr.SelectData):
 
94
  x, y = evt.index
95
  color = red if point_label_radio == 0 else blue
96
  if prompt_image is None:
@@ -112,7 +113,6 @@ with block:
112
 
113
 
114
  def on_everything_image_select(input_image, pred_masks, masks, text, evt: gr.SelectData):
115
- sam_cpu.dummy_encode(input_image)
116
  i = evt.index
117
  mask = pred_masks[i][0]
118
  print(mask)
 
91
  reset_btn.click(on_reset_btn_click, [input_image], [input_image, point_coords, point_labels], queue=False)
92
 
93
  def on_prompt_image_select(input_image, prompt_image, point_coords, point_labels, point_label_radio, text, pred_masks, embedding, evt: gr.SelectData):
94
+ sam_cpu.dummy_encode(input_image)
95
  x, y = evt.index
96
  color = red if point_label_radio == 0 else blue
97
  if prompt_image is None:
 
113
 
114
 
115
  def on_everything_image_select(input_image, pred_masks, masks, text, evt: gr.SelectData):
 
116
  i = evt.index
117
  mask = pred_masks[i][0]
118
  print(mask)