Doing prediction with CoreML

prediction.swift

func predict(image: UIImage) -> Species? {
    do {
        if let resizedImage = resize(image: image, newSize: trainedImageSize), let pixelBuffer = resizedImage.toCVPixelBuffer() {
            let model = DogAndCatCNN()
            let prediction = try model.prediction(data: pixelBuffer)
            if prediction.species[0].intValue == 1 {
                return .Dog
             } else {
                return .Cat
             }
        }
    } catch {
        print("Error while doing predictions: \(error)")
    }

    return nil
}