From 033ef80e4e721cd980bf3e0c4f0c112bab2dd603 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 24 四月 2023 11:17:24 +0800
Subject: [PATCH] Merge pull request #405 from alibaba-damo-academy/dev_aky
---
funasr/bin/punctuation_infer.py | 4 +---
1 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/punctuation_infer.py b/funasr/bin/punctuation_infer.py
index a801ee8..077814d 100644
--- a/funasr/bin/punctuation_infer.py
+++ b/funasr/bin/punctuation_infer.py
@@ -23,7 +23,7 @@
from funasr.utils import config_argparse
from funasr.utils.types import str2triple_str
from funasr.utils.types import str_or_none
-from funasr.punctuation.text_preprocessor import split_to_mini_sentence
+from funasr.datasets.preprocessor import split_to_mini_sentence
class Text2Punc:
@@ -61,7 +61,6 @@
text_name="text",
non_linguistic_symbols=train_args.non_linguistic_symbols,
)
- print("start decoding!!!")
@torch.no_grad()
def __call__(self, text: Union[list, str], split_size=20):
@@ -223,7 +222,6 @@
result, _ = text2punc(line)
item = {'key': key, 'value': result}
results.append(item)
- print(results)
return results
for inference_text, _, _ in data_path_and_name_and_type:
--
Gitblit v1.9.1