Dede16 commited on
Commit
57cd36e
·
verified ·
1 Parent(s): 311d448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import cv2
5
  from ultralytics import YOLO
6
 
7
  # Initialize the YOLO model
8
- model = YOLO('runs/segment/train/weights/best.pt')
9
 
10
  # Define directories
11
  image_dir = 'datasetsw/valid/ripe'
 
5
  from ultralytics import YOLO
6
 
7
  # Initialize the YOLO model
8
+ model = YOLO('best.pt')
9
 
10
  # Define directories
11
  image_dir = 'datasetsw/valid/ripe'