Update custom_interface.py

custom_interface.py

@@ -22,12 +22,13 @@ 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)
+        print(prediction)
+        predicted_words = prediction
         return predicted_words
 
     def filter_repetitions(self, seq, max_repetition_length):