File size: 651 Bytes
32cc554 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
from service.predict import workflow
def process_image(image):
disease_name, remedy = workflow(image)
return disease_name, remedy
# Create the Gradio interface
iface = gr.Interface(
fn=process_image,
inputs=gr.Image(
image_mode="RGB",
sources="upload",
label="Upload Plant Disease Image",
show_download_button=True,
type="pil",
),
outputs=[
gr.Textbox(label="Prediction", placeholder="Disease Prediction"),
gr.Markdown(label="Remedy"),
],
title="Classify Plant Diseases and Get Remedies",
)
if __name__ == "__main__":
iface.launch()
|