Spaces:
Running
on
Zero
Running
on
Zero
--DELETE--
#5
by
atalaydenknalbant
- opened
- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
@@ -25,7 +25,7 @@ category_dict = {
|
|
25 |
}
|
26 |
|
27 |
|
28 |
-
@spaces.GPU()
|
29 |
def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
|
30 |
model = YOLO(f"{model_id}.pt")
|
31 |
results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
|
|
|
25 |
}
|
26 |
|
27 |
|
28 |
+
@spaces.GPU(duration=200)
|
29 |
def yolov10_inference(image, model_id, image_size, conf_threshold, iou_threshold):
|
30 |
model = YOLO(f"{model_id}.pt")
|
31 |
results = model(source=image, imgsz=image_size, iou=iou_threshold, conf=conf_threshold, verbose=False)[0]
|
requirements.txt
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
supervision
|
2 |
ultralytics
|
3 |
spaces
|
4 |
gradio
|
|
|
1 |
+
supervision
|
2 |
ultralytics
|
3 |
spaces
|
4 |
gradio
|