Update text_engine.py

precommihted already
2 years ago · d2da7f50d2
parent 82f731c153
commit d2da7f50d2
1 changed files with 3 additions and 2 deletions
--- a/paddlespeech/server/engine/text/python/text_engine.py
+++ b/paddlespeech/server/engine/text/python/text_engine.py
@ -107,10 +107,11 @@ class PaddleTextConnectionHandler:
            assert len(tokens) == len(labels)

            text = ''
+            print(self._punc_list)
            for t, l in zip(tokens, labels):
                text += t
                if l != 0:  # Non punc.
-                    text += self._punc_list[l]
+                    text += self._punc_list[l - 1]

            return text
        else:
@ -160,7 +161,7 @@ class TextEngine(BaseEngine):
            return False

        self.executor = TextServerExecutor()
-        self.executor._init_from_path(
+        self.executor._init_from_path_new(
            task=config.task,
            model_type=config.model_type,
            lang=config.lang,