Update

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
3 years ago · 418e54ca03
1 changed files with 1 additions and 1 deletions
--- a/torch_vggish.py
+++ b/torch_vggish.py
@ -42,6 +42,6 @@ class TorchVggish(Operator):

    def __call__(self, audio_path: str) -> NamedTuple('Outputs', [('embs', numpy.ndarray)]):
        audio_tensors = self.model.preprocess(audio_path)
-        features = self.model._model(audio_tensors)
+        features = self.model(audio_tensors)
        Outputs = NamedTuple('Outputs', [('embs', numpy.ndarray)])
        return Outputs(features.detach().numpy())