import platform
import pathlib

if platform.system() == "Windows":
    temp = pathlib.PosixPath
    pathlib.PosixPath = pathlib.WindowsPath

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

learn = load_learner(Path("model.pkl"))

def classify_image(img):
    pred, idx, probs = learn.predict(img)

    return f"This is {pred}.\nConfidence Level: {float(max(probs))}"


image = gr.Image(shape=(192, 192))
examples = ['images/pizza.jpg', 'images/pasta.jpg', 'images/dunno.jpg']

app = gr.Interface(fn=classify_image,
                   inputs=image,
                   outputs=gr.Textbox(label="Output"),
                   examples=examples)
app.launch()