Spaces:
Sleeping
Sleeping
Rename main2.py to main.py
Browse files- 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 |
-
|
| 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)
|