diff --git a/src/scripts/image_demo.py b/src/scripts/image_demo.py
index 4c091b9..1872782 100644
--- a/src/scripts/image_demo.py
+++ b/src/scripts/image_demo.py
@@ -52,8 +52,8 @@ lut_model = LoadCheckpoint(config.lut_model_path).cuda()
 print(net_model)
 print(lut_model)
 
-lr_image = cv2.imread(str(config.project_path / "data" / "Set14/LR/X4/monarch.png"))[:,::-1,::-1].copy()
-image_gt = cv2.imread(str(config.project_path / "data" / "Set14/HR/monarch.png"))[:,::-1,::-1].copy()
+lr_image = cv2.imread(str(config.lr_image_path))[:,:,::-1].copy()
+image_gt = cv2.imread(str(config.hr_image_path))[:,:,::-1].copy()
 
 input_image = torch.tensor(lr_image).type(torch.float32).permute(2,0,1)[None,...].cuda()