Spaces:
Sleeping
Sleeping
| from tensorflow import keras | |
| import gradio as gr | |
| path_to_model = "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), | |
| theme = "darkhuggingface" | |
| ) | |
| iface.launch(debug=True) |