Spaces:
Runtime error
Runtime error
Commit
·
741f3dc
1
Parent(s):
017cea0
Add application file
Browse files- app.py +2 -2
- EyesCareEyeDetectModel.pt → best.pt +0 -0
app.py
CHANGED
|
@@ -24,8 +24,8 @@ def yoloV8_func(image: gr.Image = None,
|
|
| 24 |
iou_threshold (float, optional): Intersection over Union threshold for object detection. Defaults to 0.50.
|
| 25 |
"""
|
| 26 |
# Load the YOLOv8 model from the 'best.pt' checkpoint
|
| 27 |
-
model_path = "EyesCareEyeDetectModel.pt"
|
| 28 |
-
model = YOLO(
|
| 29 |
|
| 30 |
# Perform object detection on the input image using the YOLOv8 model
|
| 31 |
results = model.predict(image,
|
|
|
|
| 24 |
iou_threshold (float, optional): Intersection over Union threshold for object detection. Defaults to 0.50.
|
| 25 |
"""
|
| 26 |
# Load the YOLOv8 model from the 'best.pt' checkpoint
|
| 27 |
+
#model_path = "EyesCareEyeDetectModel.pt"
|
| 28 |
+
model = YOLO("best.pt")
|
| 29 |
|
| 30 |
# Perform object detection on the input image using the YOLOv8 model
|
| 31 |
results = model.predict(image,
|
EyesCareEyeDetectModel.pt → best.pt
RENAMED
|
File without changes
|