RiceLeafDisease / app.py
Abhi101's picture
Update app.py
a579bcd
raw
history blame
610 Bytes
from tensorflow import keras
import gradio as gr
path_to_model = "my_model_new_vgg.h5"
model = keras.models.load_model(path_to_model)
def classify_image(inp):
# inp = load_image(inp_path)
inp = inp.reshape((-1, 224, 224, 3))
prediction = model.predict(inp).tolist()[0]
class_names = ['Bacterialblight', 'Blast', 'Brownspot', 'Tungro']
return {class_names[i]: prediction[i] for i in range(4)}
iface = gr.Interface(fn=classify_image,
inputs=gr.inputs.Image(shape=(224, 224)),
outputs=gr.outputs.Label(num_top_classes=4),
)
iface.launch(debug=True)