Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ transform = transforms.Compose([
62
  ])
63
 
64
  # Class labels
65
- class_names = ["No Alzheimer’s", "Very Mild", "Mild", "Moderate"]
66
 
67
  # --------- Prediction Function ----------
68
  def predict(image):
 
62
  ])
63
 
64
  # Class labels
65
+ class_names = ['No Impairment', 'Very Mild Impairment', 'Moderate Impairment', 'Mild Impairment']
66
 
67
  # --------- Prediction Function ----------
68
  def predict(image):