cdcvd commited on
Commit
7111eb4
·
verified ·
1 Parent(s): 02d921d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,line_width=1)
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]