Update custom_interface.py

custom_interface.py

@@ -86,7 +86,7 @@ class CustomSLUDecoder(Pretrained):
         with torch.no_grad():
             wavs, wav_lens = wavs.to(self.device), wav_lens.to(self.device)
             encoder_out = self.encode_batch(wavs)
-            predicted_tokens, scores = self.mods.beam_searcher(
+            predicted_tokens, scores, _, _ = self.mods.beam_searcher(
                 encoder_out, wav_lens
             )
             predicted_words = [