Knight-coderr commited on
Commit
335eff1
·
verified ·
1 Parent(s): 497606d

Upload 2 files

Browse files
utils/speaker_diarization.py CHANGED
@@ -1,7 +1,11 @@
1
  from pyannote.audio import Pipeline
 
 
 
2
 
3
  diarization_pipeline = Pipeline.from_pretrained(
4
- "pyannote/speaker-diarization-3.1"
 
5
  )
6
 
7
  def diarize_speakers(audio_path):
 
1
  from pyannote.audio import Pipeline
2
+ import os
3
+
4
+ hf_token = os.environ.get("HF_TOKEN")
5
 
6
  diarization_pipeline = Pipeline.from_pretrained(
7
+ "pyannote/speaker-diarization-3.1",
8
+ use_auth_token=hf_token
9
  )
10
 
11
  def diarize_speakers(audio_path):
utils/vad_segmentation.py CHANGED
@@ -1,8 +1,10 @@
 
1
  from pyannote.audio import Model
2
  from pyannote.audio.pipelines import VoiceActivityDetection
 
3
 
4
  model = Model.from_pretrained("pyannote/segmentation",
5
- use_auth_token="")
6
 
7
 
8
  vad_pipeline = VoiceActivityDetection(segmentation=model)
 
1
+ import os
2
  from pyannote.audio import Model
3
  from pyannote.audio.pipelines import VoiceActivityDetection
4
+ hf_token = os.environ.get("HF_TOKEN")
5
 
6
  model = Model.from_pretrained("pyannote/segmentation",
7
+ use_auth_token=hf_token)
8
 
9
 
10
  vad_pipeline = VoiceActivityDetection(segmentation=model)