Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -239,7 +239,7 @@ def get_bbox_with_draw(image, img_state, evt: gr.SelectData):
|
|
239 |
)
|
240 |
return img_state, image
|
241 |
|
242 |
-
def check_and_infer_box(
|
243 |
if len(img_state_bbox.selected_bboxes) == 2:
|
244 |
return inference_box(input_image, img_state_bbox, class_names_input_box)
|
245 |
return None
|
|
|
239 |
)
|
240 |
return img_state, image
|
241 |
|
242 |
+
def check_and_infer_box(input_image, img_state_bbox,class_names_input_box):
|
243 |
if len(img_state_bbox.selected_bboxes) == 2:
|
244 |
return inference_box(input_image, img_state_bbox, class_names_input_box)
|
245 |
return None
|