python_package.md

Python Package

Install

pip install git+https://github.com/wenet-e2e/wenet.git

Development Install

git clone https://github.com/wenet-e2e/wenet.git
cd wenet
pip install -e .

Command line Usage

wenet --language chinese audio.wav

You can specify the following parameters.

• -l or --language: chinese/english are supported now.
• -m or --model_dir: your own model dir
• -g or --gpu: the device id of gpu, default value -1 represents for cpu.
• -t or --show_tokens_info: show the token level information such as timestamp, confidence, etc.
• --align: force align the input audio and transcript
• --label: the input label to align
• --paraformer: use the best Chinese model
• --device: specify the backend accelerator (cuda/npu/cpu)

Python Programming Usage

import wenet

model = wenet.load_model('chinese')
# or model = wenet.load_model(model_dir='xxx')
result = model.transcribe('audio.wav')
print(result['text'])