huffelaf/rabbitdetector

1# libraries2from fastai.vision.all import *3import gradio as gr4import pathlib5temp = pathlib.PosixPath6pathlib.PosixPath = pathlib.WindowsPath7 8# functions9def is_rabbit(x) : return x[0].isupper()10 11def classify_image(img):12    pred,pred_idx,probs = learn_inf.predict(img)13    return dict(zip(categories, map(float, probs)))14 15# load model16learn_inf = load_learner('model.pkl')17 18# categories19categories = ("rabbit", "dogs", "bears")20 21# gradio functionality22image = gr.inputs.Image(shape=(224, 224))23label = gr.outputs.Label()24examples = ["images/rabbit.jpg", "images/dog.jpg", "images/bear.jpg"]25 26intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)27intf.launch(inline=False)28 29 30