jarif's picture
Update app.py
c3471a3
raw
history blame
904 Bytes
from fastai.vision.all import *
from fastai.vision.all import load_learner
import gradio as gr
fruit_labels = ('Apple', 'Apricot', 'Avocado',
'Banana', 'Blueberry',
'Carambola', 'Cherry', 'Fig',
'Grape', 'Kiwi', 'Lemon',
'Lychee', 'Mango',
'Orange', 'Papaya',
'Pear', 'Pineapple',
'Raspberry', 'Strawberry', 'Watermelon')
model=load_learner("fruit_model_v6.pkl")
def recognize_image(image):
pred, idx, probs = model.predict(image)
print(pred)
return dict(zip(fruit_labels, map(float, probs)))
image = gr.inputs.Image(shape=(192,192))
label = gr.outputs.Label(num_top_classes=5)
examples = [
'test_0.jpg',
'test_1.jpg',
'test_2.jpg',
'test_4.jpeg'
]
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
iface.launch(inline=False)