Spaces:
Runtime error
Runtime error
Edvin Behdadijd
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,11 +5,11 @@ import cv2
|
|
| 5 |
from ultralytics import YOLO
|
| 6 |
|
| 7 |
# Load the YOLOv8 model
|
| 8 |
-
model = YOLO('best.pt')
|
| 9 |
|
| 10 |
def detect_objects(image):
|
| 11 |
# Convert the input image to a format YOLO can work with
|
| 12 |
-
image = np.array(image
|
| 13 |
|
| 14 |
# Perform detection
|
| 15 |
results = model(image)[0]
|
|
|
|
| 5 |
from ultralytics import YOLO
|
| 6 |
|
| 7 |
# Load the YOLOv8 model
|
| 8 |
+
model = YOLO('best.pt',line_width=1)
|
| 9 |
|
| 10 |
def detect_objects(image):
|
| 11 |
# Convert the input image to a format YOLO can work with
|
| 12 |
+
image = np.array(image)
|
| 13 |
|
| 14 |
# Perform detection
|
| 15 |
results = model(image)[0]
|