jarif's picture
Update app.py
b890276 verified
raw
history blame contribute delete
986 Bytes
from fastai.vision.all import *
from fastai.vision.all import load_learner
import gradio as gr
fruit_labels = ('Anthracnose', 'Apple Scab', 'Black Spot', 'Blight',
'Blossom End Rot', 'Botrytis', 'Brown Rot', 'Canker',
'Cedar Apple Rust', 'Clubroot', 'Crown Gall', 'Downy Mildew',
'Fire Blight', 'Fusarium', 'Gray Mold', 'Leaf Spots', 'Mosaic Virus',
'Nematodes', 'Powdery Mildew', 'Verticillium')
model = load_learner("crop_model_1.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) # Remove the unused parameter 'type'
examples = [
'image_1.jpg',
'image_2.jpeg',
'image_3.jpeg',
'image_4.jpeg'
]
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
iface.launch(inline=False)