FlowerRecognizer.swift

import CoreML
import Vision

class FlowerRecognizer {
    let model: VNCoreMLModel
    
    init() {
        let modelURL = Bundle.main.url(forResource: "FlowerModel", withExtension: "mlmodel")!
        model = try! VNCoreMLModel(for: MLModel(contentsOf: modelURL))
    }
    
    func recognizeFlower(image: UIImage) -> String {
        let request = VNCoreMLRequest(model: model) { request, error in
            guard let results = request.results as? [VNClassificationObservation],
                  let topResult = results.first else {
                return
            }
            
            print("Recognized flower: \(topResult.identifier)")
        }
        
        let handler = VNImageRequestHandler(cgImage: image.cgImage!, options: [:])
        try! handler.perform([request])
    }
}