Spaces:
Sleeping
Sleeping
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]
|