Update custom_interface.py

custom_interface.py

@@ -21,14 +21,13 @@ 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]
-        print(predicted_words)
         prediction = []
         for sent in predicted_words:
             sent = self.filter_repetitions(sent, 3)
-            sent = " ".join(sent)
+            # sent = " ".join(sent)
             prediction.append(sent)
-        print(prediction)
-        predicted_words = prediction
+        print([prediction])
+        predicted_words = [prediction]
         return predicted_words
 
     def filter_repetitions(self, seq, max_repetition_length):