Update app.py

app.py

@@ -16,7 +16,7 @@ def predict_alzheimer(img):
     img_array = preprocess_input(img_array)  # Preprocess the input as expected by ResNet50
     
     prediction = model.predict(img_array)  # Predict using the model
-    classes = ['Mild_Demented', 'Non_Demented', 'Very_Mild_Demented' ]  # Specific names
+    classes = ['Mild Demented', 'Non Demented', 'Very Mild Demented' ]  # Specific names
     return {classes[i]: float(prediction[0][i]) for i in range(3)}  # Return the prediction
 
 # Define Gradio interface