Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import numpy as np
-import time
+import os, time
 from pyannote.audio import Pipeline
 import librosa, torch
 from transformers import pipeline
@@ -14,8 +14,8 @@ device = 0 if torch.cuda.is_available() else "cpu"
 SAMPLE_RATE = 16_000
 
 ######## LOAD MODELS FROM HUB ########
-dia_model = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=True)
-vad_model = Pipeline.from_pretrained("pyannote/voice-activity-detection", use_auth_token=True)
+dia_model = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token=os.environ['TOKEN'])
+vad_model = Pipeline.from_pretrained("pyannote/voice-activity-detection", use_auth_token=os.environ['TOKEN'])
 pipe = pipeline(task="automatic-speech-recognition", model=MODEL_NAME, chunk_length_s=chunk_length_s, device=device)
 pipe.model.config.forced_decoder_ids = pipe.tokenizer.get_decoder_prompt_ids(language=lang, task="transcribe")