app.py

import gradio as gr
from fastai.vision.all import *

learn = load_learner('pets.pkl')

categories = ("Cat", "Dog", "Fish")

examples = ['examples/cat.jpg', 'examples/dog.jpg', 'examples/fish.jpg']

def categorize(image):
    pred,_,probs = learn.predict(image)
    return dict(zip(categories, map(float, probs)))

iface = gr.Interface(fn=categorize, inputs="image", outputs="label", examples=examples)
iface.launch(inline=False)