EagleUI/bearClassifier

1from fastai.vision.all import *2import gradio as gr3import skimage4 5 6learn_inf = load_learner('export.pkl')7 8def predict(img):9    img = PILImage.create(img)10    pred,pred_idx,probs = learn_inf.predict(img)11    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'12 13gr.Interface(fn=predict, inputs=gr.components.Image(shape=(512, 512)), outputs=gr.components.Label(num_top_classes=3)).launch(share=True)