app.py

# AUTOGENERATED! DO NOT EDIT! File to edit: ../prod.ipynb.

# %% auto 0
__all__ = ['categories', 'image', 'label', 'examples', 'intf', 'classify_image']

# %% ../prod.ipynb 1
from fastai.vision.all import * 
import gradio as gr


learner = load_learner('model.pkl')

# %% ../prod.ipynb 5
categories = ('1990s', '2010s')
def classify_image(img):
    pred, idx, probs = learner.predict(img)
    return dict(zip(categories, map(float, probs)))

# %% ../prod.ipynb 6
image = gr.Image(shape=(192,192))
label = gr.Label()
title = "1990s or 2010s classifier"
description = "A nba eras classifier trained on images from Bing Image search with fastai."
examples = ['sample.png', 'sample2.png', 'sample3.png']

intf = gr.Interface(fn=classify_image,inputs=image,outputs=label,examples=examples, title=title, description=description)
intf.launch(inline=False)