Auth

diarization.py

@@ -5,9 +5,11 @@ import re
 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", use_auth_token='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')
 
+    # Initialize diarization pipeline
+    accesstoken = os.environ['Diarization']
+    pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1", token=accesstoken )
+    
     # Send pipeline to GPU (when available)
     pipeline.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))