sdhca/flying_sails

1# AUTOGENERATED! DO NOT EDIT! File to edit: ../nbs/02_app.ipynb.2 3# %% auto 04__all__ = ['learn', 'image', 'label', 'examples', 'intf', 'classify_image']5 6# %% ../nbs/02_app.ipynb 27from fastai.vision.all import PILImage, load_learner8import gradio as gr9 10# %% ../nbs/02_app.ipynb 411learn = load_learner('models/sailboat_sails_classification.pkl')12 13# %% ../nbs/02_app.ipynb 614def classify_image(img):15    pred, idx, probs = learn.predict(img)16    return dict(zip(learn.dls.vocab, map(float, probs)))17 18# %% ../nbs/02_app.ipynb 919image = gr.Image(shape=(192, 192))20label = gr.Label(show_label=False)21examples = [f'sailboat{i}.png' for i in range(1,5)]22 23# examples24intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)25intf.launch(inline=False)26