dkenney99/minimal

1from fastai.vision.all import *2import gradio as gr3 4 5def is_cat(x):6    return x[0].isupper()7 8 9learn = load_learner('model.pkl')10 11categories = ('Dog', 'Cat')12 13 14def classify_image(img):15    pred, idx, probs = learn.predict(img)16    return dict(zip(categories, map(float, probs)))17 18 19examples = ['dog.jpg', 'cat.jpg']20 21interface = gr.Interface(22    fn=classify_image,23    inputs=gr.Image(),24    outputs=gr.Label(num_top_classes=3),25    examples=examples,26)27 28interface.launch(share=False)29