Update app.py

app.py

@@ -46,14 +46,14 @@ def prepare_dataset(extracted_folder):
     image_paths = []
     labels = []
     
-    for disease_folder in ['alzheimers_datasets', 'parkinson_datasets', 'MSjpg']:
+    for disease_folder in ['alzheimers_dataset', 'parkinsons_dataset', 'MSjpg']:
         folder_path = os.path.join(neuronii_path, disease_folder)
         
         # Check if the subfolder exists
         if not os.path.exists(folder_path):
             raise FileNotFoundError(f"The folder {disease_folder} does not exist at path: {folder_path}")
         
-        label = {'alzheimers_datasets': 0, 'parkinson_datasets': 1, 'MSjpg': 2}[disease_folder]
+        label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'MSjpg': 2}[disease_folder]
         
         for img_file in os.listdir(folder_path):
             if img_file.endswith(('.nii', '.jpg', '.jpeg')):