Spaces:

lord-reso
/

host

Sleeping

lord-reso commited on Feb 17, 2024

Commit

07c1f47

verified ·

1 Parent(s): 43db691

Update logic.py

Files changed (1) hide show

logic.py CHANGED Viewed

@@ -95,7 +95,7 @@ def load_speaker_model(speaker_model_path):
     return model
-speaker_model = load_speaker_model('speaker/saved_models/saved_model_e300.pt').to(device).eval().float()
 def extract_speech_embedding(audio_path: str):
     audio, sampling_rate = load_wav_to_torch(audio_path)

     return model
+speaker_model = load_speaker_model('speaker/saved_models/saved_model_e273_LargeBatch.pt').to(device).eval().float()
 def extract_speech_embedding(audio_path: str):
     audio, sampling_rate = load_wav_to_torch(audio_path)