Spaces:
Runtime error
Runtime error
update
Browse files
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'])),
|