From 586c7024ef1ffdf2658ae8aedaedfc4c35089491 Mon Sep 17 00:00:00 2001 From: "mengzhe.cmz" Date: Wed, 22 Mar 2023 11:21:39 +0800 Subject: [PATCH] fix demo service --- .../infer.py | 1 - funasr/bin/punctuation_infer_vadrealtime.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py b/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py index 2bac22037..baad18884 100644 --- a/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py +++ b/egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py @@ -13,7 +13,6 @@ logger.setLevel(logging.CRITICAL) inference_pipeline = pipeline( task=Tasks.punctuation, model='damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727', - model_revision="v1.0.0", output_dir="./tmp/" ) diff --git a/funasr/bin/punctuation_infer_vadrealtime.py b/funasr/bin/punctuation_infer_vadrealtime.py index d6cc15332..ea8c04120 100644 --- a/funasr/bin/punctuation_infer_vadrealtime.py +++ b/funasr/bin/punctuation_infer_vadrealtime.py @@ -69,6 +69,7 @@ class Text2Punc: precache = "".join(cache) else: precache = "" + cache = [] data = {"text": precache + text} result = self.preprocessor(data=data, uid="12938712838719") split_text = self.preprocessor.pop_split_text_data(result) @@ -233,7 +234,6 @@ def inference_modelscope( item = {'key': key, 'value': ""} results.append(item) return results - #import pdb;pdb.set_trace() result, _, cache = text2punc(line, cache) item = {'key': key, 'value': result, 'cache': cache} results.append(item)