diff --git a/model/pre.py b/model/pre.py
index f1cdeb6..96dcf74 100644
--- a/model/pre.py
+++ b/model/pre.py
@@ -47,7 +47,7 @@ def remove_module_prefix(state_dict):
         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
     ])
 
-    img_folder = "../input/testA"
-    output_folder = "../input/out"
+    img_folder = "./input/testA"
+    output_folder = "./input/out"
     predict_and_save(model, img_folder, val_transform, device, output_folder)