Spaces:

sc45
/

FOREIGN-WHISPERS

Sleeping

sc45 commited on Dec 11, 2023

Commit

ad14260

•

1 Parent(s): 124bf87

Auth

Files changed (1) hide show

diarization.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 def perform_diarization(audio_file_path, translated_file_path, output_dir='./audio/diarization'):
     # Initialize diarization pipeline
-    pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1")
     # Send pipeline to GPU (when available)
     pipeline.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))

 def perform_diarization(audio_file_path, translated_file_path, output_dir='./audio/diarization'):
     # Initialize diarization pipeline
+    pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1", use_auth_token='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')
     # Send pipeline to GPU (when available)
     pipeline.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))