paorph/figureIdentifier

1from fastai.vision.all import *2import gradio as gr3 4learn = load_learner('model.pkl')5 6categories = ('circle','square','triangle')7 8def classify_image(img):9    pred,idx, probs = learn.predict(img)10    return dict(zip(categories, map(float, probs)))11 12image  = gr.inputs.Image(shape=(192,192))13label = gr.outputs.Label()14examples = ['circle.png','square.png','triangle.png']15 16intf = gr.Interface(fn=classify_image, inputs = image, outputs = label, examples = examples)17intf.launch(inline=False)18