Mahbubur/tiger3

1from fastai.vision.all import *2import gradio as gr3 4 5 6learn = load_learner('tiger-model-v3.pkl')7categories = ('Bengal', 'Siberian', 'White')8 9def classify_img(img):10    preds, idx, probs = learn.predict(img)11    return dict(zip(categories, map(float, probs)))12 13image = gr.Image()14label = gr.Label()15examples = ['ben-tiger.jpg', 'sib-tiger.jpg']16 17iface = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)18iface.launch()19