diff --git a/bin/infer_on_folders.py b/bin/infer_on_folders.py
index e9d5f10..c44bd31 100644
--- a/bin/infer_on_folders.py
+++ b/bin/infer_on_folders.py
@@ -1,6 +1,8 @@
 import argparse
 import glob
 import os
+import sys
+sys.path.append('.')
 
 import torch
 from torchvision import transforms