Update custom_interface.py

custom_interface.py

@@ -20,7 +20,12 @@ class ASR(Pretrained):
 
         # Output layer for seq2seq log-probabilities
         predictions = self.hparams.test_search(encoded_outputs, self.wav_lens)[0]
-        predicted_words = [self.hparams.tokenizer.decode_ids(prediction).split(" ") for prediction in predictions if 0 not in prediction]
+        # 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 = []
         for sent in predicted_words:
             sent = self.filter_repetitions(sent, 3)