Update custom_interface.py

custom_interface.py

@@ -23,9 +23,8 @@ class ASR(Pretrained):
         # predicted_words = [self.hparams.tokenizer.decode_ids(prediction).split(" ") for prediction in predictions]
         predicted_words = []
         for prediction in predictions:
-            print(prediction)
-            if prediction != 0:
-                predicted_words.append(self.hparams.tokenizer.decode_ids(prediction).split(" "))
+            prediction = [token for token in prediction if token != 0]
+            predicted_words.append(self.hparams.tokenizer.decode_ids(prediction).split(" "))
         prediction = []
         for sent in predicted_words:
             sent = self.filter_repetitions(sent, 3)