Spaces:
Running
Running
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) |