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


model = load_learner("model.pkl")

def predict(img):
    prediction, idx, confidence = model.predict(img)
    return prediction


image = gr.inputs.Image(shape=(200, 200), source="webcam")
label = gr.outputs.Label()

interface = gr.Interface(fn=predict, inputs=image, outputs=label)
interface.launch()