Spaces:
Runtime error
Runtime error
File size: 637 Bytes
3d03641 6399b4c 3d03641 5a6f5a0 3d03641 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | __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)
|