Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,6 @@ category_dict = {0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H',
|
|
| 26 |
|
| 27 |
@spaces.GPU
|
| 28 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
| 29 |
-
# Download models
|
| 30 |
|
| 31 |
|
| 32 |
model = YOLO(model_id)
|
|
|
|
| 26 |
|
| 27 |
@spaces.GPU
|
| 28 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
|
|
|
| 29 |
|
| 30 |
|
| 31 |
model = YOLO(model_id)
|