Mahmoud Elsamadony commited on
Commit
b5282ea
·
1 Parent(s): 2d6ec47
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,10 +22,10 @@ WHISPER_COMPUTE_TYPE = os.environ.get("WHISPER_COMPUTE_TYPE", "int8_float32")
22
 
23
  # Diarization: download locally to avoid repeated API calls
24
  DIARIZATION_REPO_ID = os.environ.get(
25
- "DIARIZATION_REPO_ID", "pyannote/speaker-diarization-3.1"
26
  )
27
  DIARIZATION_LOCAL_DIR = os.environ.get(
28
- "DIARIZATION_LOCAL_DIR", "models/pyannote-speaker-diarization-3.1"
29
  )
30
 
31
  HF_TOKEN = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACEHUB_API_TOKEN")
 
22
 
23
  # Diarization: download locally to avoid repeated API calls
24
  DIARIZATION_REPO_ID = os.environ.get(
25
+ "DIARIZATION_REPO_ID", "Revai/reverb-diarization-v2"
26
  )
27
  DIARIZATION_LOCAL_DIR = os.environ.get(
28
+ "DIARIZATION_LOCAL_DIR", "models/Revai-reverb-diarization-v2"
29
  )
30
 
31
  HF_TOKEN = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACEHUB_API_TOKEN")