mircomarahrens/gallus-classifier

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('export.pkl')5 6categories = learn.dls.vocab7def classify_image(img):8    pred,pred_idx,probs = learn.predict(img)9    return dict(zip(categories, map(float, probs)))10 11image = gr.inputs.Image(shape=(512, 512))12label = gr.outputs.Label()13 14examples = ['chabo.jpg', 'pekin.jpeg', 'brahma.jpeg']15 16intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)17intf = intf.launch(inline=False)