whisper-speaker-diarization

Runtime error

sanchit-gandhi commited on Jan 1, 2024

Commit

659a76f

1 Parent(s): 99975a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,11 +11,11 @@ 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,
 )
 def tuple_to_string(start_end_tuple, ndigits=1):

 device = 0 if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.environ.get("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):