Spaces:

Gokulavelan
/

audio-speaker-diarization

Sleeping

Gokulavelan commited on Dec 7, 2024

Commit

ff239c3

1 Parent(s): ea5aa75

test

Files changed (1) hide show

config.py CHANGED Viewed

@@ -30,7 +30,7 @@ class InferenceConfig(BaseModel):
 model_settings_data = {
     "DIARIZATION_MODEL": "pyannote/speaker-diarization-3.1",
     "HF_TOKEN": os.environ.get("HF_TOKEN"),
-    "ASR_MODEL": "openai/whisper-large-v3",
     "ASSISTANT_MODEL": "distil-whisper/distil-large-v3"
 }

 model_settings_data = {
     "DIARIZATION_MODEL": "pyannote/speaker-diarization-3.1",
     "HF_TOKEN": os.environ.get("HF_TOKEN"),
+    "ASR_MODEL": "openai/whisper-small",
     "ASSISTANT_MODEL": "distil-whisper/distil-large-v3"
 }