rohb/animal_classification

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('model.plk')5categories = ("Amphibian", "Bird", "Fish", "Invertebrate", "Mammal", "Reptile")6 7def classify_image(img):8    pred, idx, prob = learn.predict(img)9    return dict(zip(categories, map(float, prob)))10 11image = gr.Image(height=256, width=256)12label = gr.Label()13examples = ['amphibian.jpg', 'bird.jpg', 'fish.jpg', 'invertebrate.jpg', 'mammal.jpg', 'reptile.jpg']14 15intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)16intf.launch(inline=False)