Update app.py

app.py

@@ -39,7 +39,7 @@ def preprocess_image(image_path):
 def prepare_dataset(extracted_folder):
     image_paths = []
     labels = []
-    for disease_folder in ['alzheimers', 'parkinsons', 'ms']:
+    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'ms']:
         folder_path = os.path.join(extracted_folder, disease_folder)
         label = {'alzheimers': 0, 'parkinsons': 1, 'ms': 2}[disease_folder]
         for img_file in os.listdir(folder_path):