Update app.py

app.py

@@ -51,7 +51,7 @@ def prepare_dataset(extracted_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}")
+            print(f"The folder {disease_folder} does not exist at path: {folder_path}")
         
         label = {'alzheimers_dataset': 0, 'parkinsons_dataset': 1, 'MSjpg': 2}[disease_folder]
         
@@ -59,8 +59,10 @@ def prepare_dataset(extracted_folder):
             if img_file.endswith(('.nii', '.jpg', '.jpeg')):
                 image_paths.append(os.path.join(folder_path, img_file))
                 labels.append(label)
-    
-    return image_paths, labels
+            else:
+                print(f"Unsuported file:{img_file}")
+        print(f"Total images loaded :{len(image_paths)}")
+        return image_paths, labels
 
 # Custom Dataset class
 class CustomImageDataset(Dataset):