diff --git a/config.py b/config.py
index 82eea53..5dad7ea 100644
--- a/config.py
+++ b/config.py
@@ -28,7 +28,7 @@
     'collate_fn': get_collate_fn,
 
     # test data
-    'test_data': 'data/test_data/'
+    'test_data': 'data/example_data/'
 }
 
 
@@ -47,7 +47,7 @@
         'input_channel': 64,
         'channels': [1024, 512, 256, 128, 64], # channels for deconvolutional layers
         'output_channel': 3, # images with RGB channels
-        'model_path': 'models/generator.pth',
+        'model_path': 'pretrained_models/generator.pth',
     },
     # FACE EMBEDDING NETWORK (f)
     'f': {