Hugging Face's logo

Spaces:
vyluong
/
PoC_ASR_v6_dev
Running

App Files Community
vyluong commited on
Commit
1fe53e1
·
verified ·
1 Parent(s): 04bd887

Update app/core/config.py

Browse files
Files changed (1) hide show
  1. app/core/config.py +4 -1
app/core/config.py CHANGED
@@ -49,7 +49,10 @@ class Settings(BaseSettings):
49
  # Diarization model
50
  # pyannote/speaker-diarization-3.1
51
  # pyannote/speaker-diarization-community-1
52
- diarization_model: str = "pyannote/speaker-diarization-community-1"
 
 
 
53
 
54
  # Device settings
55
  device: Literal["cuda", "cpu", "auto"] = "auto"
 
49
  # Diarization model
50
  # pyannote/speaker-diarization-3.1
51
  # pyannote/speaker-diarization-community-1
52
+ pyannote_model: str = "pyannote/speaker-diarization-community-1"
53
+ sortformer_model = "nvidia/diar_sortformer_4spk-v1"
54
+
55
+ diarization_backend = "sortformer"
56
 
57
  # Device settings
58
  device: Literal["cuda", "cpu", "auto"] = "auto"