Merlintxu commited on
Commit
40aba40
·
verified ·
1 Parent(s): fe4ae7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -15,6 +15,9 @@ from pyannote.audio import Pipeline
15
  warnings.filterwarnings("ignore")
16
  logging.set_verbosity_error()
17
 
 
 
 
18
  # Updated models by language
19
  MODELS = {
20
  "es": [
@@ -61,7 +64,7 @@ def detect_language(audio_path):
61
  return max(langs, key=lambda x: x.prob).lang
62
 
63
  def diarize_audio(wav_audio):
64
- pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization")
65
  diarization = pipeline(wav_audio)
66
  return diarization
67
 
 
15
  warnings.filterwarnings("ignore")
16
  logging.set_verbosity_error()
17
 
18
+ # Read the Hugging Face token from the environment variable
19
+ HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
20
+
21
  # Updated models by language
22
  MODELS = {
23
  "es": [
 
64
  return max(langs, key=lambda x: x.prob).lang
65
 
66
  def diarize_audio(wav_audio):
67
+ pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=HUGGINGFACE_TOKEN)
68
  diarization = pipeline(wav_audio)
69
  return diarization
70