Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ category_dict = {0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H',
|
|
| 20 |
|
| 21 |
|
| 22 |
|
| 23 |
-
@spaces.GPU
|
| 24 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
| 25 |
model_path = download_models(model_id)
|
| 26 |
if model_id[:7] == 'yolov10':
|
|
|
|
| 20 |
|
| 21 |
|
| 22 |
|
| 23 |
+
@spaces.GPU
|
| 24 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
| 25 |
model_path = download_models(model_id)
|
| 26 |
if model_id[:7] == 'yolov10':
|