wondervictor commited on
Commit
e53c1e6
·
verified ·
1 Parent(s): 3dd984b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(img_state_bbox, input_image, 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
 
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