Edvin Behdadijd commited on
Commit
02d921d
·
verified ·
1 Parent(s): a8f2bb2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ 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]
 
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]