Spaces:
Sleeping
Sleeping
Commit
•
4a5d0fb
1
Parent(s):
ffa2e5d
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,6 +32,7 @@ category_dict = {
|
|
| 32 |
}
|
| 33 |
|
| 34 |
|
|
|
|
| 35 |
def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
|
| 36 |
model_path = download_models(model_id)
|
| 37 |
model = YOLOv10(model_path)
|
|
|
|
| 32 |
}
|
| 33 |
|
| 34 |
|
| 35 |
+
@spaces.GPU
|
| 36 |
def yolov10_inference(image, video, model_id, image_size, conf_threshold, iou_threshold):
|
| 37 |
model_path = download_models(model_id)
|
| 38 |
model = YOLOv10(model_path)
|