Spaces:
Runtime error
Runtime error
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)
|