Spaces:
Sleeping
Sleeping
import gradio as gr | |
from fastai import load_learner | |
learner = load_learner('model.pkl') | |
def classify_images(img): | |
pred, index, probs = learner.predict(img) | |
return {learner.dls.vocab[i]: float(probs[i]) for i in range(len(learner.dls.vocab))} | |
intface = gr.Interface(fn=classify_images, inputs=gr.inputs.Image(type='pil'), | |
outputs=gr.outputs.Label(num_top_classes=2), | |
title="Kentish plover vs White-faced plover Classifier", | |
description="Upload an image of either Kentish plover and White-faced plover." | |
) | |
intface.launch(share=True) |