File size: 526 Bytes
34ccbbc 5fef537 34ccbbc df45c31 d2971cb 34ccbbc df45c31 d2971cb 34ccbbc a505967 df45c31 a505967 34ccbbc c018a0a 34ccbbc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import gradio as gr
import numpy as np
import pandas as pd
from tensorflow import keras
model = keras.models.load_model('model.h5')
classes = ['Mild_Demented', 'Moderate_Demented', 'Non_Demented', 'Very_Mild_Demented']
def analyse(image):
data = image.reshape((1, 128, 128, 3))
predict = model.predict(data)[0]
return classes[np.argmax(predict)]
iface = gr.Interface(
analyse,
gr.inputs.Image(shape=(128,128)),
"text",
examples=['non.jpg', 'moderate.jpg', 'mild.jpg', 'verymild.jpg'],
)
iface.launch() |