azzusCode/fantasy-creatures-classifier

1from fastai.vision.all import *2import gradio as gr3 4fantasy_creatures = (5    'Angel', 'Basilisk', 'Centaur', 'Chimera', 'Cyclops', 'Demon', 'Dragon', 'Dwarf', 'Elf', 'Fairy', 6    'Fire Elemental', 'Genie', 'Ghost', 'Goblin', 'Golem', 'Griffin', 'Hydra', 'Kraken', 'Lich', 'Mermaid', 7    'Minotaur', 'Orc', 'Pegasus', 'Phoenix', 'Unicorn', 'Water Elemental', 'Werewolf', 'Wizard', 'Wraith', 'Zombie'8)9 10model = load_learner("models/creatures-classifier-v2.pkl")11 12def recognize_image(image):13    pred, idx, probs = model.predict(image)14    return dict(zip(fantasy_creatures, map(float, probs)))15 16image = gr.Image()17label = gr.Label(num_top_classes=5)18examples = [19    'test_images/image1.jpg',20    'test_images/image2.jpg',21    'test_images/image3.jpg',22    'test_images/image4.jpg',23    'test_images/image5.jpg'24    ]25 26iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)27iface.launch(inline=False)28