add models

export-onnx.py

@@ -105,12 +105,14 @@ def main():
         license_url = (
             "https://huggingface.co/pyannote/segmentation-3.0/blob/main/LICENSE"
         )
+        model_author = "pyannote-audio"
     else:
         url_1 = "https://huggingface.co/Revai/reverb-diarization-v1"
         url_2 = "https://huggingface.co/csukuangfj/sherpa-onnx-reverb-diarization-v1"
         license_url = (
             "https://huggingface.co/Revai/reverb-diarization-v1/blob/main/LICENSE"
         )
+        model_author = "Revai"
 
     meta_data = {
         "num_speakers": len(model.specifications.classes),
@@ -122,7 +124,7 @@ def main():
         "receptive_field_shift": receptive_field_shift,
         "model_type": "pyannote-segmentation-3.0",
         "version": "1",
-        "model_author": "pyannote",
+        "model_author": model_author,
         "maintainer": "k2-fsa",
         "url_1": url_1,
         "url_2": url_2,

model.int8.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eba77d4d983fd37d8cc0864b10b63bcd0c1864633b2917fb0d2aeb4b6055cf23
-size 2415969
+oid sha256:cd4576325e0703478da874105ac2de53180cd6220b608074281c19f849535f52
+size 2415966

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:08a5eb1dff82847656baa7bf022c85bd9c90dd5a6a1f1b06f3ebfca0482beb5a
-size 9512160
+oid sha256:c1b1628a86d12ea9837c2feb0eab3b08ba95111670813d26f09b75b6dbb7540b
+size 9512157

run.sh

@@ -3,13 +3,19 @@
 
 export SHERPA_ONNX_IS_REVAI=1
 
+if [ -z $VERSION ]; then
+  VERSION=v1
+fi
+
 set -ex
 function install_pyannote() {
-  pip install pyannote.audio onnx onnxruntime
+  pip install pyannote.audio onnx onnxruntime numpy==1.26.4
 }
 
 function download_test_files() {
-  curl -SL -O https://huggingface.co/Revai/reverb-diarization-v1/resolve/main/pytorch_model.bin
+  # 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
 }
 
@@ -28,13 +34,14 @@ echo "----------onnx model.onnx----------"
 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-v1/resolve/main/LICENSE
+# 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-v1/tree/main
+https://huggingface.co/Revai/reverb-diarization-$VERSION/tree/main
 
 Note that it is accessible under a non-commercial license.