Upload 2 files
Browse files- api.py +0 -0
- audio_tools.py +9 -1
api.py
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
audio_tools.py
CHANGED
|
@@ -369,9 +369,17 @@ def diarize_audio(
|
|
| 369 |
|
| 370 |
class VoiceEmbedder:
|
| 371 |
def __init__(self):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 372 |
self.model = SpeakerRecognition.from_hparams(
|
| 373 |
source="speechbrain/spkrec-ecapa-voxceleb",
|
| 374 |
-
savedir=
|
| 375 |
)
|
| 376 |
self.model.eval()
|
| 377 |
|
|
|
|
| 369 |
|
| 370 |
class VoiceEmbedder:
|
| 371 |
def __init__(self):
|
| 372 |
+
# Guardar el model de SpeechBrain en una ruta de dades escrivible
|
| 373 |
+
# En entorns com HF Spaces, el cwd pot ser de només lectura, per això
|
| 374 |
+
# fem servir /data/pretrained_models.
|
| 375 |
+
from pathlib import Path as _P
|
| 376 |
+
|
| 377 |
+
safe_savedir = _P("/data/pretrained_models/spkrec-ecapa-voxceleb")
|
| 378 |
+
safe_savedir.mkdir(parents=True, exist_ok=True)
|
| 379 |
+
|
| 380 |
self.model = SpeakerRecognition.from_hparams(
|
| 381 |
source="speechbrain/spkrec-ecapa-voxceleb",
|
| 382 |
+
savedir=str(safe_savedir),
|
| 383 |
)
|
| 384 |
self.model.eval()
|
| 385 |
|