Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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)
|