audio (#8134)

Signed-off-by: Nikolay Karpov <[email protected]> Co-authored-by: Nikolay Karpov <[email protected]>
NVIDIA · Jan 8, 2024 · f082947 · f082947
1 parent 9526584
commit f082947
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/nemo/collections/asr/models/label_models.py b/nemo/collections/asr/models/label_models.py
@@ -525,9 +525,7 @@ def get_label(
         np.random.seed(random_seed)
         starts = np.random.randint(0, audio_length - duration + 1, size=num_segments)
         for start in starts:
-            audio = audio[start : start + duration]
-
-            _, logits = self.infer_segment(audio)
+            _, logits = self.infer_segment(audio[start : start + duration])
             label_id = logits.argmax(axis=1)
             label_id_list.append(int(label_id[0]))