Update config.py

Author: n0obcoder

config.py

@@ -1,13 +1,13 @@
-pkl_dir_path = 'pickles'
-train_val_df_pkl_path = 'train_val_df.pickle'
-test_df_pkl_path = 'test_df.pickle'
+pkl_dir_path             = 'pickles'
+train_val_df_pkl_path    = 'train_val_df.pickle'
+test_df_pkl_path         = 'test_df.pickle'
 disease_classes_pkl_path = 'disease_classes.pickle'
-models_dir = 'models'
-
+models_dir               = 'models'
 
 from torchvision import transforms
 normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
                                  std=[0.229, 0.224, 0.225])
+
 # transforms.RandomHorizontalFlip() not used because some disease might be more likely to the present in a specific lung (lelf/rigth)
 transform = transforms.Compose([transforms.ToPILImage(), 
                     transforms.Resize(224),