|
|
|
|
|
from fastcore.all import * |
|
from fastai.vision.all import * |
|
from fastbook import load_learner |
|
import gradio as gr |
|
|
|
inf = load_learner('LeopardCheetah.pkl') |
|
|
|
categories = ('Cheetah','Images','Leopard') |
|
|
|
def classify_img(img): |
|
pred, idx, probs = inf.predict(img) |
|
return dict(zip(categories,map(float,probs))) |
|
|
|
|
|
|
|
|
|
|
|
image = gr.Image(shape=(192,192)) |
|
label = gr.outputs.Label() |
|
|
|
examples = ['cheetah.jpg','leopard.jpg','img.jpg'] |
|
|
|
intf = gr.Interface(fn = classify_img, inputs=image,outputs=label,examples = examples) |
|
|
|
intf.launch(inline=False) |
|
|
