from fastai.vision.all import load_learner
import gradio as gr

# Load the model
model_path = "convnext_model/convnext_model.pkl"
learn = load_learner(model_path)

# Prediction function
def predict(image):
    pred, _, probs = learn.predict(image)
    return pred

# Gradio interface
iface = gr.Interface(
    fn=predict,
    inputs="image",
    outputs="label",
    title="Greens classifier",
    description="Dill, Parsley, Coriander or Watercress?"
)

iface.launch()