diff --git a/code/data.py b/code/data.py
index 40c4d7685dc5066341638b993eda5c138cef8040..aafb73234b283dc83b83b436ea81f37c938e5b76 100644
--- a/code/data.py
+++ b/code/data.py
@@ -123,12 +123,12 @@ class TestDataModule(pl.LightningDataModule):
 
             #For single channel
             #For JSC Training.
-            #pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
-            #cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
+            pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
+            cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
 
             #For Local Machine Training.
-            pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
-            cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
+            #pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
+            #cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
             
             pli_files_list = [file for file in os.listdir(pli_path) if
                               file.endswith(('.h5', '.hdf', '.h4', '.hdf4', '.he2', '.hdf5', '.he5'))]