import SwiftUI

struct ImageDetailView: View {
   @State private var className: String = ""
   let imageName: String

   var body: some View {
      VStack {
         Image(imageName)
            .resizable()
            .scaledToFit()
            .frame(width: 200, height: 200)
            .padding()
         Button("Identify") {
            let provider = ModelProvider.shared
            let prediction = provider.prediction(image: imageName)
            className = prediction
         }.buttonStyle(.borderedProminent)
         .padding()
         Text(className)
         Spacer()
      }
   }
}
#Preview {
   ImageDetailView(imageName: ModelProvider.shared.images.first!)
}