diff --git a/run_CNN.py b/run_CNN.py
index cdeca01e501766c8770b91786d648022b2e305df..218c066d5de187011bd9bc58af40ce4b68ace34c 100644
--- a/run_CNN.py
+++ b/run_CNN.py
@@ -58,7 +58,7 @@ class Dataset(torch.utils.data.Dataset):
 
 # prepare model
 model = models.get[args.specie]['archi']
-model.load_state_dict(torch.load(f"{os.path.dirname(__file__)}/weights/{models.get[args.specie]['weights']}"))
+model.load_state_dict(torch.load(f".{os.path.dirname(__file__)}/weights/{models.get[args.specie]['weights']}"))
 model.eval()
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model.to(device)