speedosteveo/first

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('model.pkl')5 6# Align with learn.dls.vocab!7categories = ('Cat', 'Dog')8 9def classify_image(img):10    pred,idx,probs = learn.predict(img)11    return dict(zip(categories, map(float, probs)))12 13 14image = gr.inputs.Image(shape=(192, 192))15label = gr.outputs.Label()16examples = ['dog.jpg', 'cat.jpg', 'Donkey.jpg']17 18intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)19intf.launch(inline=False, share=True)