mohammedafeef commited on
Commit
a374eb5
·
verified ·
1 Parent(s): 2c7769c

Rename main2.py to main.py

Browse files
Files changed (1) hide show
  1. main2.py → main.py +4 -2
main2.py → main.py RENAMED
@@ -66,9 +66,11 @@ class UnifiedAudioAnalyzer:
66
 
67
  # Load pyannote diarization pipeline
68
  try:
 
 
69
  self.diarization_pipeline = Pipeline.from_pretrained(
70
- "pyannote/speaker-diarization-3.1"
71
- # Uncomment and add your token: use_auth_token="YOUR_HUGGINGFACE_TOKEN"
72
  )
73
  if torch.cuda.is_available():
74
  self.diarization_pipeline = self.diarization_pipeline.to(self.device)
 
66
 
67
  # Load pyannote diarization pipeline
68
  try:
69
+ import os
70
+ hf_token = os.getenv("HF_TOKEN")
71
  self.diarization_pipeline = Pipeline.from_pretrained(
72
+ "pyannote/speaker-diarization-3.1",
73
+ use_auth_token=hf_token
74
  )
75
  if torch.cuda.is_available():
76
  self.diarization_pipeline = self.diarization_pipeline.to(self.device)