Arafath10 commited on
Commit
88236d1
·
verified ·
1 Parent(s): c6b737e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -5,7 +5,7 @@ from ultralytics import YOLO
5
  model = YOLO('best.pt') # load a custom model
6
 
7
  # Predict with the model
8
- results = model('bus.jpg')
9
  print(results)# predict on an image
10
 
11
 
 
5
  model = YOLO('best.pt') # load a custom model
6
 
7
  # Predict with the model
8
+ results = model('test.jpg')
9
  print(results)# predict on an image
10
 
11