diff --git a/distill_and_select.py b/distill_and_select.py
index 07cc379..52c3bea 100644
--- a/distill_and_select.py
+++ b/distill_and_select.py
@@ -18,9 +18,9 @@ from torchvision import transforms
 from towhee.operator.base import NNOperator
 from towhee import register
 from PIL import Image as PILImage
-from model.feature_extractor import FeatureExtractor
-from model.students import FineGrainedStudent, CoarseGrainedStudent
-from model.selector import SelectorNetwork
+from .model.feature_extractor import FeatureExtractor
+from .model.students import FineGrainedStudent, CoarseGrainedStudent
+from .model.selector import SelectorNetwork
 from torch import nn