| | |
|---|
| | | |
|---|
| | | TORCH_DIR=$(python3 -c "import torch; import os; print(os.path.dirname(torch.__file__))") |
|---|
| | | BLADE_DIR=$(python3 -c "import torch_blade; import os; print(os.path.dirname(torch_blade.__file__))") |
|---|
| | | export LD_LIBRARY_PATH=/usr/local/lib:${TORCH_DIR}/lib:${BLADE_DIR}:${LD_LIBRARY_PATH} |
|---|
| | | |
|---|
| | | download_model_dir="/workspace/models" |
|---|
| | | model_dir="damo/speech_paraformer-large-vad-punc_asr_nat-zh-cn-16k-common-vocab8404-onnx" |
|---|
| | | model_dir="damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-onnx" |
|---|
| | | vad_dir="damo/speech_fsmn_vad_zh-cn-16k-common-onnx" |
|---|
| | | punc_dir="damo/punc_ct-transformer_cn-en-common-vocab471067-large-onnx" |
|---|
| | | itn_dir="thuduj12/fst_itn_zh" |
|---|
| | |
|---|
| | | port=10095 |
|---|
| | | certfile="$(pwd)/ssl_key/server.crt" |
|---|
| | | keyfile="$(pwd)/ssl_key/server.key" |
|---|
| | | hotword="$(pwd)/hotwords.txt" |
|---|
| | | hotword="$(pwd)/websocket/hotwords.txt" |
|---|
| | | # set decoder_thread_num |
|---|
| | | decoder_thread_num=$(cat /proc/cpuinfo | grep "processor"|wc -l) || { echo "Get cpuinfo failed. Set decoder_thread_num = 32"; decoder_thread_num=32; } |
|---|
| | | multiple_io=16 |
|---|
| | |
|---|
| | | --port ${port} \ |
|---|
| | | --certfile "${certfile}" \ |
|---|
| | | --keyfile "${keyfile}" \ |
|---|
| | | --hotword "${hotword}" |
|---|
| | | --gpu \ |
|---|
| | | --hotword "${hotword}" & |
|---|
| | | |
|---|
