Update custom_interface.py

custom_interface.py

@@ -22,12 +22,10 @@ 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 = []
-        print(predicted_words)
         for sent in predicted_words:
             sent = self.filter_repetitions(sent, 3)
             prediction.append(sent)
         predicted_words = prediction
-        print(predicted_words)
         return predicted_words
 
     def filter_repetitions(self, seq, max_repetition_length):
@@ -94,6 +92,3 @@ class ASR(Pretrained):
         outputs = self.encode_batch(batch, rel_length)
        
         return outputs
-
-    # def forward(self, wavs, wav_lens=None):
-    #     return self.encode_batch(wavs=wavs, wav_lens=wav_lens)