diff --git a/imageClasification/imageClasification.py b/imageClasification/imageClasification.py
index 705e0c8..f32473a 100644
--- a/imageClasification/imageClasification.py
+++ b/imageClasification/imageClasification.py
@@ -11,7 +11,7 @@ from tensorflow.keras.models import Sequential
 batch_size = 32
 img_height = 180
 img_width = 180
-data_dir = "../SI_InteligentnyWozekWidlowy/imageClasification/images"
+data_dir = "./images"
 
 train_ds = tf.keras.utils.image_dataset_from_directory(
     data_dir,