Spaces:
Sleeping
Sleeping
StevenChen16
commited on
Commit
•
3983c11
1
Parent(s):
14d3c09
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ category_dict = {
|
|
31 |
77: 'teddy bear', 78: 'hair drier', 79: 'toothbrush'
|
32 |
}
|
33 |
|
34 |
-
@spaces.GPU(duration=
|
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)
|
|
|
31 |
77: 'teddy bear', 78: 'hair drier', 79: 'toothbrush'
|
32 |
}
|
33 |
|
34 |
+
@spaces.GPU(duration=120)
|
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)
|