app.py

# AUTOGENERATED! DO NOT EDIT! File to edit: 02a_prod_artsie.ipynb.

# %% auto 0
__all__ = ['repo_model_name', 'remote_learner', 'labels', 'demo', 'classify_img']

# %% 02a_prod_artsie.ipynb 4
from fastai.vision.all import *
from huggingface_hub import push_to_hub_fastai, from_pretrained_fastai
import gradio as gr

# %% 02a_prod_artsie.ipynb 37
repo_model_name = 'artsie-model'

# %% 02a_prod_artsie.ipynb 40
remote_learner = from_pretrained_fastai(f"rahuketu86/{repo_model_name}"); remote_learner

# %% 02a_prod_artsie.ipynb 47
labels = remote_learner.dls.vocab

def classify_img(img):
    img = PILImage.create(img)
    pred, pred_idx, probs = remote_learner.predict(img)
    return {labels[i]:float(probs[i]) for i in range(len(labels))}

# %% 02a_prod_artsie.ipynb 50
demo = gr.Interface(fn=classify_img, 
                    inputs=gr.Image(), 
                    outputs=gr.Label(num_top_classes=len(labels)))

# %% 02a_prod_artsie.ipynb 52
demo.launch(inline=False)