dctellya/minimal-docker

1# AUTOGENERATED! DO NOT EDIT! File to edit: ../dogs-vs-cats.ipynb.2 3# %% auto 04__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'intf', 'is_cat', 'classify_image']5 6# %% ../dogs-vs-cats.ipynb 27from fastai.vision.all import *8import gradio as gr9 10def is_cat(x): 11    return x[0].isupper()12 13# %% ../dogs-vs-cats.ipynb 414learn = load_learner('model.pkl')15 16# %% ../dogs-vs-cats.ipynb 617categories = ('Dog', 'Cat')18 19def classify_image(img):20    pred,idx,probs = learn.predict(img)21    return dict(zip(categories,map(float,probs)))22    23 24# %% ../dogs-vs-cats.ipynb 825image = gr.Image(height=192,width=192)26label = gr.Label(label="output")27examples = ['dog.jpg','cat.jpg','dunno.jpg']28 29intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)30intf.launch(server_name="0.0.0.0", server_port=7860)31 32 33