Spaces:
Runtime error
Runtime error
Commit
·
02d10dd
1
Parent(s):
1d10fb9
update
Browse files- app_locally.py +1 -1
app_locally.py
CHANGED
@@ -20,7 +20,7 @@ def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
|
|
20 |
|
21 |
tts_model = TTS(language=language, device=device)
|
22 |
speaker_id = tts_model.hps.data.spk2id
|
23 |
-
speaker_key =
|
24 |
source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)
|
25 |
|
26 |
|
|
|
20 |
|
21 |
tts_model = TTS(language=language, device=device)
|
22 |
speaker_id = tts_model.hps.data.spk2id
|
23 |
+
speaker_key = speaker_id.lower().replace('_', '-')
|
24 |
source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)
|
25 |
|
26 |
|