Spaces:
Build error
Build error
Nguyen Thai Thao Uyen
commited on
Commit
·
bd723e6
1
Parent(s):
1079c88
Add application file
Browse files
app.py
CHANGED
|
@@ -14,10 +14,10 @@ torch.hub.download_url_to_file(
|
|
| 14 |
'three.jpg')
|
| 15 |
|
| 16 |
|
| 17 |
-
def yoloV8_func(image: gr.
|
| 18 |
-
image_size: gr.
|
| 19 |
-
conf_threshold: gr.
|
| 20 |
-
iou_threshold: gr.
|
| 21 |
"""This function performs YOLOv8 object detection on the given image.
|
| 22 |
|
| 23 |
Args:
|
|
|
|
| 14 |
'three.jpg')
|
| 15 |
|
| 16 |
|
| 17 |
+
def yoloV8_func(image: gr.Image = None,
|
| 18 |
+
image_size: gr.Slider = 640,
|
| 19 |
+
conf_threshold: gr.Slider = 0.4,
|
| 20 |
+
iou_threshold: gr.Slider = 0.50):
|
| 21 |
"""This function performs YOLOv8 object detection on the given image.
|
| 22 |
|
| 23 |
Args:
|