cocktailpeanut commited on
Commit
02d10dd
·
1 Parent(s): 1d10fb9
Files changed (1) hide show
  1. 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 = speaker_key.lower().replace('_', '-')
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