csukuangfj's picture
add models
11788ff
#!/usr/bin/env bash
# Copyright 2024 Xiaomi Corp. (authors: Fangjun Kuang)
export SHERPA_ONNX_IS_REVAI=1
if [ -z $VERSION ]; then
VERSION=v1
fi
set -ex
function install_pyannote() {
pip install pyannote.audio onnx onnxruntime numpy==1.26.4
}
function download_test_files() {
# curl -SL -O https://huggingface.co/Revai/reverb-diarization-$VERSION/resolve/main/pytorch_model.bin
curl -SL -O https://huggingface.co/openspeech/revai-models/resolve/main/$VERSION/pytorch_model.bin
curl -SL -O https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/lei-jun-test.wav
}
install_pyannote
download_test_files
./export-onnx.py
./preprocess.sh
echo "----------torch----------"
./vad-torch.py
echo "----------onnx model.onnx----------"
./vad-onnx.py --model ./model.onnx --wav ./lei-jun-test.wav
echo "----------onnx model.int8.onnx----------"
./vad-onnx.py --model ./model.int8.onnx --wav ./lei-jun-test.wav
# curl -SL -O https://huggingface.co/Revai/reverb-diarization-$VERSION/resolve/main/LICENSE
curl -SL -O https://huggingface.co/openspeech/revai-models/resolve/main/v1/LICENSE
cat >README.md << EOF
# Introduction
Models in this file are converted from
https://huggingface.co/Revai/reverb-diarization-$VERSION/tree/main
Note that it is accessible under a non-commercial license.
Please see ./LICENSE for details.
See also
https://www.rev.com/blog/speech-to-text-technology/introducing-reverb-open-source-asr-diarization
EOF