ContentView.swift

//
//  ContentView.swift
//  Mac-PaliGemma2
//
//  Created by KUNAL BATRA on 12/22/24.
//

import SwiftUI
import MLXModelManager

struct ContentView: View {
    @StateObject var manager = ModelManager(modelPath: "mlx-community/paligemma2-3b-ft-docci-448-8bit")
    
    var body: some View {
        VStack {
            Button("Describe Image"){
                Task {
                    //load model
                    try await manager.loadModel()
                    
                    //inference
                    await manager.generate(
                        prompt: "Describe this image.",
                        imagePath:"/Users/kunalbatra/Downloads/classic.png"
                    )
                }
            }
            Text(manager.output)
        }
        .padding()
    }
}

#Preview {
    ContentView()
}