paorph/animalClassifier

1# -*- coding: utf-8 -*-2 3 4from fastai.vision.all import *5import gradio as gr6 7 8learn = load_learner('model.pkl')9 10 11categories = learn.dls.vocab12 13def classify_image(img):14    pred,idx,probs = learn.predict(img)15    return dict(zip(categories, map(float,probs)))16 17 18image = gr.inputs.Image(shape=(192, 192))19label = gr.outputs.Label()20examples = ['chimpanzee.png', 'gibbon.png', 'gorilla.png', 'human.png','orangutan.png']21 22intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)23intf.launch(inline=False)24