Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from fastai.vision.all import * | |
| import hashlib | |
| import numpy as np | |
| from PIL import Image | |
| learn = load_learner("model-034-concorde-barlett-3.pkl") | |
| categories = ["Barlett", "Concorde"] | |
| def classify_pear(pear): | |
| pred, idx, probs = learn.predict(pear) | |
| result = f"This is a {pred} pear.\n\n{dict(zip(categories, map(float, probs)))}" | |
| return result | |
| demo = gr.Interface(fn=classify_pear, inputs="image", outputs="text") | |
| demo.launch() | |