dkenney99/ctv

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('export.pkl')5 6categories = ('Sedan', 'Truck', 'Van')7 8 9def classify_image(img):10    pred, idx, probs = learn.predict(img)11    return dict(zip(categories, map(float, probs)))12 13 14examples = ['sedan.jpg', 'truck.jpg', 'van.jpg']15 16interface = gr.Interface(17    fn=classify_image,18    inputs=gr.Image(),19    outputs=gr.Label(num_top_classes=3),20    examples=examples,21)22 23interface.launch(share=False)24