diff --git a/launch.sh b/launch.sh
index f73fccdb5c71f0e0628dffd776a83a713f6e89df..a31ce092ca9d01a6fbb6948254e2e69f993c7af8 100644
--- a/launch.sh
+++ b/launch.sh
@@ -1,4 +1,4 @@
 #!/bin/sh
-dir="data_split"
+dir="/media/DATA/AZZA/splits"
 
-python main.py $dir/train_X_pxl_$1.npy $dir/train_X_median_$1.npy $dir/train_y_$1.npy $dir/valid_X_pxl_$1.npy $dir/valid_X_median_$1.npy $dir/valid_y_$1.npy modelTwoBranch $1
+python main.py $dir/train_X_pxl_$1.npy $dir/train_X_median_$1.npy $dir/train_gt_$1.npy $dir/valid_X_pxl_$1.npy $dir/valid_X_median_$1.npy $dir/valid_gt_$1.npy $dir/test_X_pxl_$1.npy $dir/test_X_median_$1.npy $dir/test_gt_$1.npy modelTwoBranch $1