From d6ebdbc8d34a1028cb773ea62be1a57bcfaffdc7 Mon Sep 17 00:00:00 2001 From: Filip Haltmayer Date: Wed, 16 Feb 2022 14:30:39 -0800 Subject: [PATCH] update Signed-off-by: Filip Haltmayer --- torchaudio_audio_embedding.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchaudio_audio_embedding.py b/torchaudio_audio_embedding.py index f835b59..b9e3af6 100644 --- a/torchaudio_audio_embedding.py +++ b/torchaudio_audio_embedding.py @@ -17,8 +17,8 @@ class TorchaudioAudioEmbedding(Operator): self._bundle = getattr(torchaudio.pipelines, name) self._model = self._bundle.get_model() - def __call__(self, image_file: 'str') -> NamedTuple('Outputs', [('embedding', numpy.ndarray)]): - waveform, sample_rate = torchaudio.load(image_file) + def __call__(self, audio_path: 'str') -> NamedTuple('Outputs', [('embedding', numpy.ndarray)]): + waveform, sample_rate = torchaudio.load(audio_path) waveform = torchaudio.functional.resample(waveform, sample_rate, self._bundle.sample_rate) embedding, _ = self._model.extract_features(waveform) embedding = embedding[0].detach().numpy()