Using a pre-trained YOLO model for object detection