Spaces:
Runtime error
Runtime error
## libraries for data preprocessing | |
import numpy as np | |
import pandas as pd | |
## libraries for training dl models | |
import tensorflow as tf | |
from tensorflow import keras | |
## libraries for pre-trained neural network | |
from tensorflow.keras.applications.xception import preprocess_input | |
## libraries for loading batch images | |
from tensorflow.keras.preprocessing.image import load_img | |
import gradio as gr | |
def get_y(o): | |
return [parent_label(o)] | |
## lets load the model | |
model = keras.models.load_model('xception_v1_15_0.812.h5') | |
def maize_disease_classifier(image): | |
x = np.array(image) | |
X = np.array([x]) | |
X = preprocess_input(X) | |
pred = model.predict(X) | |
result = pred[0].argmax() | |
## lets create our labels | |
labels = { | |
0: 'maize ear rot', | |
1: 'maize fall armyworm', | |
2: 'maize stem borer' | |
} | |
label = labels[pred[0].argmax()] | |
return pred, result, label | |
iface = gr.Interface(fn=maize_disease_classifier, inputs=gr.inputs.Image(shape=(224, 224)), \ | |
outputs=["number", "number", "text"]) | |
iface.launch(inline=False) |