fix the bug of import ctc decoder

3 years ago · 1aef1f0858
parent b453b425af
commit 1aef1f0858
1 changed files with 3 additions and 3 deletions
--- a/deepspeech/modules/ctc.py
+++ b/deepspeech/modules/ctc.py
@ -23,9 +23,9 @@ from deepspeech.utils.log import Log
 logger = Log(__name__).getlog()

 try:
-    from deepspeech.decoders.swig_wrapper import ctc_beam_search_decoder_batch  # noqa: F401
-    from deepspeech.decoders.swig_wrapper import ctc_greedy_decoder  # noqa: F401
-    from deepspeech.decoders.swig_wrapper import Scorer  # noqa: F401
+    from deepspeech.decoders.ctcdecoder.swig_wrapper import ctc_beam_search_decoder_batch  # noqa: F401
+    from deepspeech.decoders.ctcdecoder.swig_wrapper import ctc_greedy_decoder  # noqa: F401
+    from deepspeech.decoders.ctcdecoder.swig_wrapper import Scorer  # noqa: F401
 except Exception as e:
    logger.info("ctcdecoder not installed!")