| import gradio as gr | |
| from fastai.learner import load_learner | |
| from fastai.vision.core import PILImage | |
| from huggingface_hub import hf_hub_download | |
| learner = load_learner(hf_hub_download("jdmartinev/isitabird","model.pkl")) | |
| def classify_image(image): | |
| img = PILImage.create(image) | |
| pred_class, pred_idx, outputs = learner.predict(img) | |
| return(f"Predicted class: {pred_class}") | |
| demo = gr.Interface(classify_image, gr.Image(shape=(224, 224)), "text") | |
| gr.Interface(fn=classify_image, | |
| inputs=gr.Image(width=224, height=224), | |
| outputs="textbox").launch() | |