add sherpa-onnx-reverb-diarization-v1

model.py

@@ -63,13 +63,21 @@ def get_file(
 
 
 def get_speaker_segmentation_model(repo_id) -> str:
- assert repo_id in ("pyannote/segmentation-3.0",)
+ assert repo_id in (
+ "pyannote/segmentation-3.0",
+ "Revai/reverb-diarization-v1",
+ )
 
  if repo_id == "pyannote/segmentation-3.0":
  return get_file(
  repo_id="csukuangfj/sherpa-onnx-pyannote-segmentation-3-0",
  filename="model.onnx",
  )
+ elif repo_id == "Revai/reverb-diarization-v1":
+ return get_file(
+ repo_id="csukuangfj/sherpa-onnx-reverb-diarization-v1",
+ filename="model.onnx",
+ )
 
 
 def get_speaker_embedding_model(model_name) -> str:
@@ -122,7 +130,10 @@ def get_speaker_diarization(
  return sherpa_onnx.OfflineSpeakerDiarization(config)
 
 
-speaker_segmentation_models = ["pyannote/segmentation-3.0"]
+speaker_segmentation_models = [
+ "pyannote/segmentation-3.0",
+ "Revai/reverb-diarization-v1",
+]
 
 
 nemo_speaker_embedding_models = [