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

Merge branch 'dev' of github.com:alibaba-damo-academy/FunASR into dev

Browse Source
This commit is contained in:
志浩 2023-02-03 17:38:17 +08:00
commit e046529dc0
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
funasr/bin/asr_inference_launch.py
View File

@ -223,6 +223,31 @@ def inference_launch(**kwargs):
logging.info("Unknown decoding mode: {}".format(mode)) logging.info("Unknown decoding mode: {}".format(mode))
return None return None
def inference_launch_funasr(**kwargs):
if 'mode' in kwargs:
mode = kwargs['mode']
else:
logging.info("Unknown decoding mode.")
return None
if mode == "asr":
from funasr.bin.asr_inference import inference
return inference(**kwargs)
elif mode == "uniasr":
from funasr.bin.asr_inference_uniasr import inference
return inference(**kwargs)
elif mode == "paraformer":
from funasr.bin.asr_inference_paraformer import inference
return inference(**kwargs)
elif mode == "paraformer_vad_punc":
from funasr.bin.asr_inference_paraformer_vad_punc import inference
return inference(**kwargs)
elif mode == "vad":
from funasr.bin.vad_inference import inference
return inference(**kwargs)
else:
logging.info("Unknown decoding mode: {}".format(mode))
return None
def main(cmd=None): def main(cmd=None):
print(get_commandline_args(), file=sys.stderr) print(get_commandline_args(), file=sys.stderr)
@ -251,7 +276,7 @@ def main(cmd=None):
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = gpuid os.environ["CUDA_VISIBLE_DEVICES"] = gpuid
inference_launch(**kwargs) inference_launch_funasr(**kwargs)
if __name__ == "__main__": if __name__ == "__main__":