Spaces:
Sleeping
Sleeping
File size: 610 Bytes
046c9b0 a579bcd 046c9b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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)
|