diff --git a/video_swin_transformer.py b/video_swin_transformer.py index bf6947c..1d3e89f 100644 --- a/video_swin_transformer.py +++ b/video_swin_transformer.py @@ -8,7 +8,7 @@ import numpy from towhee import register from towhee.operator.base import NNOperator from towhee.types.video_frame import VideoFrame -from towhee.models.utils.video_transforms import transform_video +from towhee.models.utils.video_transforms import transform_video, get_configs from towhee.models.video_swin_transformer import video_swin_transformer from .get_configs import configs log = logging.getLogger()