speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix demo service

Browse Source
This commit is contained in:
mengzhe.cmz 2023-03-22 11:21:39 +08:00
parent 1f0f44bd07
commit 586c7024ef
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs_modelscope/punctuation/punc_ct-transformer_zh-cn-common-vadrealtime-vocab272727/infer.py
View File

@ -13,7 +13,6 @@ logger.setLevel(logging.CRITICAL)
inference_pipeline = pipeline( inference_pipeline = pipeline(
task=Tasks.punctuation, task=Tasks.punctuation,
model='damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727', model='damo/punc_ct-transformer_zh-cn-common-vad_realtime-vocab272727',
model_revision="v1.0.0",
output_dir="./tmp/" output_dir="./tmp/"
) )

2
funasr/bin/punctuation_infer_vadrealtime.py
View File

@ -69,6 +69,7 @@ class Text2Punc:
precache = "".join(cache) precache = "".join(cache)
else: else:
precache = "" precache = ""
cache = []
data = {"text": precache + text} data = {"text": precache + text}
result = self.preprocessor(data=data, uid="12938712838719") result = self.preprocessor(data=data, uid="12938712838719")
split_text = self.preprocessor.pop_split_text_data(result) split_text = self.preprocessor.pop_split_text_data(result)
@ -233,7 +234,6 @@ def inference_modelscope(
item = {'key': key, 'value': ""} item = {'key': key, 'value': ""}
results.append(item) results.append(item)
return results return results
#import pdb;pdb.set_trace()
result, _, cache = text2punc(line, cache) result, _, cache = text2punc(line, cache)
item = {'key': key, 'value': result, 'cache': cache} item = {'key': key, 'value': result, 'cache': cache}
results.append(item) results.append(item)