completed loading function

utils/data_utils.py

@@ -272,7 +272,7 @@ def load(file_paths, orientation):
     # Do we need min-max normalization here? Will need to check when debuggint and testing
     volume = (volume - np.min(volume)) / (np.max(volume) - np.min(volume))
 
-    volume, label_map = 
+    volume, label_map = set_orientation(volume, label_map, orientation)
 
     return volume, label_map, nifty_volume.header