Update custom_interface.py

custom_interface.py

@@ -22,12 +22,12 @@ class ASR(Pretrained):
         predictions = self.hparams.test_search(encoded_outputs, self.wav_lens)[0]
         predicted_words = [self.hparams.tokenizer.decode_ids(prediction).split(" ") for prediction in predictions]
 
-        prediction = []
-        for sent in predicted_words:
-            sent = self.filter_repetitions(sent, 3)
-            sent = " ".join(sent)
-            prediction.append(sent)
-        predicted_words = prediction[0]
+        # prediction = []
+        # for sent in predicted_words:
+        #     sent = self.filter_repetitions(sent, 3)
+        #     sent = " ".join(sent)
+        #     prediction.append(sent)
+        # predicted_words = prediction[0]
         return predicted_words
 
     def filter_repetitions(self, seq, max_repetition_length):