Update app.py

app.py

@@ -50,8 +50,9 @@ def prepare_dataset(extracted_folder):
         folder_path = os.path.join(neuronii_path, disease_folder)
         
         # Check if the subfolder exists
-        if not os.path.exists(folder_path):
+          if not os.path.exists(folder_path):
             print(f"Folder not found: {folder_path}")
+            continue  # Skip this folder if it's not foun
         label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'MSjpg': 2}[disease_folder]
         
         for img_file in os.listdir(folder_path):
@@ -60,8 +61,8 @@ def prepare_dataset(extracted_folder):
                 labels.append(label)
             else:
                 print(f"Unsuported file:{img_file}")
-        print(f"Total images loaded :{len(image_paths)}")
-        return image_paths, labels
+    print(f"Total images loaded :{len(image_paths)}")
+    return image_paths, labels
 
 # Custom Dataset class
 class CustomImageDataset(Dataset):