ikuinen99 commited on
Commit
411cfb5
·
1 Parent(s): 4abd59d
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -211,6 +211,7 @@ def select_fn(image, ground_img, entity_state, evt: gr.SelectData):
211
  if label is None:
212
  return ground_img
213
  print('select_fn ==> current state: ', entity_state)
 
214
  if 'grounding' not in entity_state:
215
  ground_img, local_results = grounding_dino.prompt2mask2(image,
216
  ','.join(map(lambda x: x[0], entity_state['tags'])),
 
211
  if label is None:
212
  return ground_img
213
  print('select_fn ==> current state: ', entity_state)
214
+ torch.cuda.synchronize()
215
  if 'grounding' not in entity_state:
216
  ground_img, local_results = grounding_dino.prompt2mask2(image,
217
  ','.join(map(lambda x: x[0], entity_state['tags'])),