prediction.swift

func recognizeWithLocalModel(_ image: UIImage) {
    if let data = generateMultiArrayFrom(image: image) {
        guard let modelOutput = try? singleNumberModel?.prediction(input: data) else {
            return
        }
            
        if let result = modelOutput?.classLabel {
            self.addLabelForOCR(text: "\(result)")
        } else {
            print("no result available")
        }
    }
}