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'])),
|