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()