nosson/code-classifier

1import gradio as gr2from fastai.vision.all import *3 4learner = load_learner('export.pkl')5categories = learner.dls.vocab6 7def classify_image(img):8  pred,pred_idx,probs = learner.predict(img)9  return dict(zip(categories, map(float, probs)))10 11image = gr.inputs.Image(shape=(192,192))12label = gr.outputs.Label()13intf = gr.Interface(fn=classify_image, inputs=image, outputs=label)14 15intf.launch(inline=False)