SamSV/CatDog

1__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'intf', 'is_cat', 'classify_image']2 3# %% app.ipynb 24from fastai.vision.all import *5import gradio as gr6 7def is_cat(x): return x[0].isupper()8 9# %% app.ipynb 410learn = load_learner('model.pkl')11 12# %% app.ipynb 613categories = ('Dog', 'Cat')14 15def classify_image(img):16    pred, idx, probs = learn.predict(img)17    return dict(zip(categories, map(float, probs)))18 19# %% app.ipynb 820from gradio.components import Image, Label21 22# %% app.ipynb 923image =  Image(shape=(192,192))24label = Label()25examples = ['dog.jpeg', 'cat.jpg']26intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)27intf.launch(inline=False)