VeuReu commited on
Commit
a0fdb85
·
verified ·
1 Parent(s): d6068d0

Upload 2 files

Browse files
Files changed (2) hide show
  1. api.py +0 -0
  2. 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="pretrained_models/spkrec-ecapa-voxceleb",
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