File size: 691 Bytes
2c9865a 59b3fa0 2c9865a 59b3fa0 2c9865a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | # AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
# %% auto 0
__all__ = ['learner', 'labels', 'examples', 'predict']
# %% app.ipynb 1
from fastai.vision.all import *
# %% app.ipynb 2
learner = load_learner('export.pkl')
# %% app.ipynb 5
labels = learner.dls.vocab
def predict(img):
img = PILImage.create(img)
pred,pred_idx,probs = learner.predict(img)
return {labels[i]: float(probs[i]) for i in range(len(labels))}
# %% app.ipynb 6
examples = ['dog.jpeg', 'cat.jpeg']
import gradio as gr
gr.Interface(fn=predict,
inputs=gr.inputs.Image(shape=(512, 512)),
outputs=gr.outputs.Label(num_top_classes=3),
examples=examples).launch()
|