Update main.py

main.py

@@ -5,7 +5,7 @@ from ultralytics import YOLO
 model = YOLO('best.pt')  # load a custom model
 
 # Predict with the model
-results = model('bus.jpg') 
+results = model('test.jpg') 
 print(results)# predict on an image