wondervictor commited on
Commit
7fa46c5
·
verified ·
1 Parent(s): fec2205

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -145,7 +145,7 @@ mask_adapter = None
145
  @spaces.GPU
146
  @torch.no_grad()
147
  @torch.autocast(device_type="cuda", dtype=torch.float32)
148
- def inference_box(input_img, img_state,):
149
  if len(img_state.selected_bboxes) != 2:
150
  return None
151
  mp.set_start_method("spawn", force=True)
 
145
  @spaces.GPU
146
  @torch.no_grad()
147
  @torch.autocast(device_type="cuda", dtype=torch.float32)
148
+ def inference_box(input_img, img_state,class_names_input):
149
  if len(img_state.selected_bboxes) != 2:
150
  return None
151
  mp.set_start_method("spawn", force=True)