diff --git a/video_swin_transformer.py b/video_swin_transformer.py
index 3ee49ba..a831908 100644
--- a/video_swin_transformer.py
+++ b/video_swin_transformer.py
@@ -10,6 +10,7 @@ from towhee.operator.base import NNOperator
 from towhee.types.video_frame import VideoFrame
 from towhee.models.utils.video_transforms import transform_video, get_configs
 from towhee.models.video_swin_transformer import video_swin_transformer
+from towhee.models.video_swin_transformer.get_configs import configs
 log = logging.getLogger()
 
 
@@ -41,6 +42,7 @@ class VideoSwinTransformer(NNOperator):
         self.model_name = model_name
         self.skip_preprocess = skip_preprocess
         self.topk = topk
+        self.model_configs = configs(self.model_name)
         if classmap is None:
             class_file = os.path.join(str(Path(__file__).parent), self.model_configs['labels_file_name'])
             with open(class_file, 'r') as f: