| from fastai.data.all import * | |
| from fastai.vision.all import * | |
| from fastcore.all import * | |
| import gradio as gr | |
| def label_func(fname): | |
| return "cat" if fname.name[0].isupper() else "dog" | |
| learn=load_learner('/content/drive/MyDrive/model.plk') | |
| categories=('Dog','Cat') | |
| def classify_image(img): | |
| pred,idx,prob=learn.predict(img) | |
| return dict(zip(categories,map(float,prob))) | |
| intf=gr.Interface(fn=classify_image,inputs=gr.inputs.Image(shape=(225,225)),outputs=gr.outputs.Label()) | |
| intf.launch(inline=False) |