tomweks/fontsrecognizer

1# AUTOGENERATED! DO NOT EDIT! File to edit: ../app.ipynb.2 3# %% auto 04__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'intf', 'classify_image']5 6# %% ../app.ipynb 27from fastai.vision.all import *8import gradio as gr9 10# %% ../app.ipynb 411learn = load_learner('model.pkl')12 13# %% ../app.ipynb 614categories = ('Garamond', 'Helvetica')15 16def classify_image(img):17  pred, idx, probs = learn.predict(img)18  return dict(zip(categories, map(float, probs)))19 20# %% ../app.ipynb 821image = gr.inputs.Image(shape=(192, 192))22label = gr.outputs.Label()23examples = ['garamond.png', 'helvetica.png']24 25intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)26intf.launch(inline=False)27