Spaces:
Sleeping
Sleeping
File size: 461 Bytes
317305e bdc78b6 317305e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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()
|