diff --git a/dan/ocr/decoder.py b/dan/ocr/decoder.py index 496d9b2e357caa290a16b3c7adbd1f5a4924a26c..644f8d3b5692ec71c1f51b2c3328e2c86d55c3bd 100644 --- a/dan/ocr/decoder.py +++ b/dan/ocr/decoder.py @@ -495,7 +495,7 @@ class CTCLanguageDecoder: } self.index_to_token = {i: token for token, i in self.tokens_to_index.items()} self.blank_token_id = self.tokens_to_index[self.mapping.ctc.encoded] - + # Torchaudio's decoder # https://pytorch.org/audio/master/generated/torchaudio.models.decoder.ctc_decoder.html self.decoder = ctc_decoder(