__all__ = ['learn','classify_image','categories','image','label','examples','myinterface']
#add comments with this #

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

learn = load_learner('elf_export.pkl')
categories = ('dragonborne','dwarf','elf','halfling','human','tiefling')

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

image = gr.inputs.Image (shape = (192,192))
label = gr.outputs.Label ()
examples = ['tiefling.jpg']

myinterface = gr.Interface(fn = classify_image,inputs = image,outputs = label, examples = examples)
myinterface.launch (inline=False)