File size: 1,159 Bytes
aab9f00 409a942 a9152c1 aab9f00 409a942 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gradio as gr
from fastai.vision.all import *
import skimage
learn = load_learner('model.pkl')
labels = learn.dls.vocab
def predict(img):
img = PILImage.create(img)
pred,pred_idx,probs = learn.predict(img)
return {labels[i]: float(probs[i]) for i in range(len(labels))}
#title = "Irish Potato and Common beans diseases classfier"
#description = "An app for Irish Potato and Common beans diseases Classisfication"
#article="<p style='text-align: center'>This app detects and categorizes diseases in Irish Potatoes and Common Beans, such as Common Bean Anthracnose, Bean Rust, Early Blight, and Late Blight. If the app labels an image as <b>other</b>, it implies the uploaded image is not of a leaf, or the camera angle makes identification challenging for the model. Please refer to the examples and upload a clear, close-up picture of the leaf.</p>"
examples = ['image2.jpeg', 'image3.jpg']
interpretation='default'
enable_queue=True
gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(512, 512)),outputs=gr.outputs.Label(num_top_classes=3),examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch(share=True)
|