moro23's picture
Create app.py
a8ce70c
raw
history blame
No virus
1.09 kB
## 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)