Update voice_analysis.py

voice_analysis.py

@@ -22,7 +22,14 @@ def diarize_speakers(audio_path):
     diarization = pipeline(audio_path)
     return diarization
 
-def get_speaker_embeddings(audio_path, diarization, model):
+def get_speaker_embeddings(audio_path, diarization, model_name="pyannote/embedding"):
+    hf_token = os.environ.get("py_annote_hf_token")
+    
+    if not hf_token:
+        raise ValueError("py_annote_hf_token environment variable is not set. Please check your Hugging Face Space's Variables and secrets section.")
+
+    model = Model.from_pretrained(model_name, use_auth_token=hf_token)
+    
     waveform, sample_rate = torchaudio.load(audio_path)
     embeddings = []