Running the pre-trained model on the COCO dataset