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


learn = load_learner("model.pkl")

categories = ('Guitar','Piano')

def classify_image(image):
  outcome,idx,probs = learn.predict(image)
  return dict(zip(categories,map(float,probs)))


iface = gr.Interface(
    fn = classify_image,
    inputs = "image",
    outputs = gr.outputs.Label()
    )
iface.launch()