Update app.py

app.py

@@ -4,16 +4,18 @@ import os
 import gradio as gr
 import pytube as pt
 from speechbox import ASRDiarizationPipeline
+from huggingface_hub import login
 
 MODEL_NAME = "openai/whisper-small"
 
 device = 0 if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
+login(HF_TOKEN)
+
 pipe = ASRDiarizationPipeline.from_pretrained(
     asr_model=MODEL_NAME,
     device=device,
-    use_auth_token=HF_TOKEN,
 )
 
 def tuple_to_string(start_end_tuple, ndigits=1):