diff --git a/data.py b/data.py
index ae0dbd1e9119af5456e9d8a50981588dfa9e9e37..9070ab8949c73db8610c873aea5cb2283369760d 100644
--- a/data.py
+++ b/data.py
@@ -330,8 +330,9 @@ def getDataSplit(genNew=False, split=.8, modelClass='cat'):
 def getDataSplit_Fruits(genNew=True, split=.8, modelClass='cat'):
     # access training data, get jpeg, mask, label filenames split into training / validation sets
     if genNew:  # create split
-        basePathTraining = f'/home/mahmoud/pvnet_method/Generated_Worlds_/Generated_Worlds_Training/{modelClass}'  # os.path.dirname(os.path.realpath(__file__)) + '/LINEMOD/' + modelClass
-        basePathEvaluating = f'/home/mahmoud/pvnet_method/Generated_Worlds_/Generated_Worlds_Evaluating/{modelClass}'
+        basePathTraining = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Training/' + modelClass
+        basePathEvaluating = os.path.dirname(os.path.realpath(__file__)) + '/Generated_Worlds_/Generated_Worlds_Evaluating/' + modelClass
+        
         masterList_Training = getMasterList_Fruits(basePathTraining)
         masterList_Evalution = getMasterList_Fruits(basePathEvaluating)
         random.shuffle(masterList_Training)