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)