Update app.py

app.py

@@ -34,13 +34,13 @@ import gradio as gr
 model_cnn = tf.keras.models.load_model("CNN_binary")
 
 # define the labels for the binary classification model
-labels_cnn = {1: 'healthy', 0: 'Patients'}
+labels_cnn = {0: 'Healthy', 1: 'Patients'}
 
 # load the EfficientNet binary classification model
 model_efn = tf.keras.models.load_model("efficientNet_binary")
 
 # define the labels for the binary classification model
-labels_efn = {0: 'healthy', 1: 'Patients'}
+labels_efn = {0: 'Healthy', 1: 'Patients'}
 
 def classify_cnn(inp):
     inp = inp.reshape((-1, 224, 224, 3))