diff --git a/vggish.py b/vggish.py index 4a91cc5..cf4b6f8 100644 --- a/vggish.py +++ b/vggish.py @@ -27,7 +27,7 @@ from towhee.operator.base import NNOperator from towhee.models.vggish.torch_vggish import VGG from towhee import register -import .vggish_input +import vggish_input warnings.filterwarnings('ignore') log = logging.getLogger() diff --git a/vggish_input.py b/vggish_input.py index 1bc2a13..56e6176 100644 --- a/vggish_input.py +++ b/vggish_input.py @@ -21,8 +21,8 @@ import torch import numpy as np import resampy -import .mel_features -import .vggish_params +import mel_features +import vggish_params import torchaudio