AlonBBar's picture
Save changes
c2eb29b
raw
history blame
382 Bytes
import gradio as gr
from model import predict
iface = gr.Interface(
fn=predict,
inputs=gr.Image(type="pil"),
outputs=gr.Label(num_top_classes=1),
title="Plant Disease Classifier",
description="Upload a leaf image and the model predicts the disease."
)
iface.launch(share=True)
iface.launch(capture_session=True)
if __name__ == "__main__":
iface.launch()