From 418e54ca03c613488e22247a5ad4431c214d4d47 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 9 Feb 2022 16:16:07 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- torch_vggish.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_vggish.py b/torch_vggish.py index f1b67a3..ad85496 100644 --- 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())