nut-recognizer / app.py
NasrinRipa
test image file name error
d47be8b
raw
history blame
No virus
1.03 kB
from fastai.vision.all import *
import gradio as gr
from pathlib import Path
#import pathlib
#temp = pathlib.PosixPath
#pathlib.PosixPath = pathlib.WindowsPath
nut_labels = [
'raw Almonds',
'raw Brazil nut',
'raw Cashew nut',
'raw Chestnut',
'raw Ginkgo nut',
'raw Hazelnuts',
'raw Hickory nut',
'raw Maccademia nut',
'raw Peanut',
'raw Pecans',
'raw Pili nut',
'raw Pine nut',
'raw Pistachios nut',
'raw Walnuts'
]
model = load_learner('nut-recognizer-v14 .pkl')
def recognize_image(image):
pred, idx, probs = model.predict(image)
return dict(zip(nut_labels, map(float, probs)))
image = gr.inputs.Image(shape=(192,192))
label = gr.outputs.Label(num_top_classes=5)
examples = [
'pistachios.jpeg',
'cashew.jpg',
'macadamia.jpg',
'walnuts.jpg',
'chestnuts.jpg',
'pili nuts.jpg',
'ginkgo nuts.jpg'
]
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
iface.launch(inline=False)