Spaces:
Runtime error
Runtime error
Commit
·
5d37f7b
1
Parent(s):
30a9308
Update app.py
Browse files
app.py
CHANGED
@@ -3,9 +3,9 @@ import torch
|
|
3 |
from ultralyticsplus import YOLO, render_result
|
4 |
|
5 |
|
6 |
-
torch.hub.download_url_to_file("img1.jpg", 'one.jpg')
|
7 |
-
torch.hub.download_url_to_file("img2.jpg", 'two.jpg')
|
8 |
-
torch.hub.download_url_to_file("img3.jpg", 'three.jpg')
|
9 |
|
10 |
def yoloV8_func(image: gr.inputs.Image = None,
|
11 |
image_size: gr.inputs.Slider = 640,
|
@@ -56,9 +56,9 @@ outputs = gr.outputs.Image(type="filepath", label="Output Image")
|
|
56 |
title = "YOLOv8 101: Custom Object Detection on Construction Workers"
|
57 |
|
58 |
|
59 |
-
examples = [['
|
60 |
-
['
|
61 |
-
['
|
62 |
|
63 |
yolo_app = gr.Interface(
|
64 |
fn=yoloV8_func,
|
|
|
3 |
from ultralyticsplus import YOLO, render_result
|
4 |
|
5 |
|
6 |
+
#torch.hub.download_url_to_file("img1.jpg", 'one.jpg')
|
7 |
+
#torch.hub.download_url_to_file("img2.jpg", 'two.jpg')
|
8 |
+
#torch.hub.download_url_to_file("img3.jpg", 'three.jpg')
|
9 |
|
10 |
def yoloV8_func(image: gr.inputs.Image = None,
|
11 |
image_size: gr.inputs.Slider = 640,
|
|
|
56 |
title = "YOLOv8 101: Custom Object Detection on Construction Workers"
|
57 |
|
58 |
|
59 |
+
examples = [['img1.jpg', 640, 0.5, 0.7],
|
60 |
+
['img2.jpg', 800, 0.5, 0.6],
|
61 |
+
['img13.jpg', 900, 0.5, 0.8]]
|
62 |
|
63 |
yolo_app = gr.Interface(
|
64 |
fn=yoloV8_func,
|