SohamP/animal_classification

1 2from fastai.vision.all import *3import gradio as gr4 5learn = load_learner('animal_classifier.pkl')6categories = ('Ape', 'Bear', 'Bird', 'Bison', 'Cat', 'Chicken', 'Cow', 'Deer', 'Dog', 'Dolphin', 'Duck', 'Eagle', 'Fish', 'Wolf', 'Horse', 'Lion', 'Lobster', 'Monkey', 'Pig', 'Pony', 'Rabbit', 'Shark', 'Snake', 'Spider', 'Turkey')7def classify_image(img):8    pred, i, probs = learn.predict(img)9    return dict(zip(categories, map(float, probs)))10 11image = gr.inputs.Image(shape=(192,192))12label = gr.outputs.Label()13 14examples = ['doggo2.jpg', 'cat.jpg', 'bear.jpg', 'eagle.jpg', 'lion.jpg', 'bison.jpg']15 16intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)17intf.launch(inline=False)18