import SwiftUI
import CoreML

class ModelProvider {
   var model: MyImageClassifier?
   let images = ["preview01", "preview02", "preview03", "preview04", "preview05", "preview06", "preview07", "preview08", "preview09", "preview10"]

   static let shared = ModelProvider()
   private init() {
      let config = MLModelConfiguration()
      self.model = try? MyImageClassifier(configuration: config)
   }
   func prediction(image: String) -> String {
      var className = "Not Sure"
      if let model = model, let input = convertImage(image: image) {
         do {
            let prediction = try model.prediction(image: input)
            className = prediction.target
         } catch {
            print(error)
         }
      }
      return className
   }
}