import gradio as gr
from seefood_classifier_import import get_seefood_classifier
import PIL.Image
import numpy as np

classifier = get_seefood_classifier()


def classify(image):
    im = PIL.Image.fromarray(np.array(image, dtype=np.uint8))
    preds = classifier.predict_image(im)
    results = {pred["label"]: pred["confidence"] for pred in preds}
    return results


iface = gr.Interface(
    fn=classify,
    inputs="image",
    outputs="label",
    description="Upload your favourite food pic",
)
iface.launch()