Update app.py
Browse files
app.py
CHANGED
|
@@ -50,24 +50,24 @@ def clone_and_speak(text, speaker_wav):
|
|
| 50 |
# Use English speaker model
|
| 51 |
model = TTS(language="EN", device=device)
|
| 52 |
speaker_ids = model.hps.data.spk2id
|
| 53 |
-
|
| 54 |
|
| 55 |
for speaker_key in speaker_ids.keys():
|
| 56 |
speaker_id = speaker_ids[speaker_key]
|
| 57 |
speaker_key = speaker_key.lower().replace('_', '-')
|
| 58 |
|
| 59 |
# Generate base TTS voice
|
| 60 |
-
|
| 61 |
#source_se = torch.load(f'checkpoints/base_speakers/EN/{speaker_key}.pth', map_location=device)
|
| 62 |
-
|
| 63 |
|
| 64 |
# Use speaker_wav as reference to extract style embedding
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
|
| 69 |
# Run the tone conversion
|
| 70 |
-
|
| 71 |
audio_src_path=tmp_melo_path,
|
| 72 |
src_se=ref_se,
|
| 73 |
tgt_se=ref_se,
|
|
|
|
| 50 |
# Use English speaker model
|
| 51 |
model = TTS(language="EN", device=device)
|
| 52 |
speaker_ids = model.hps.data.spk2id
|
| 53 |
+
default_speaker_id = next(iter(speaker_ids.values()))
|
| 54 |
|
| 55 |
for speaker_key in speaker_ids.keys():
|
| 56 |
speaker_id = speaker_ids[speaker_key]
|
| 57 |
speaker_key = speaker_key.lower().replace('_', '-')
|
| 58 |
|
| 59 |
# Generate base TTS voice
|
| 60 |
+
speed = 1.0
|
| 61 |
#source_se = torch.load(f'checkpoints/base_speakers/EN/{speaker_key}.pth', map_location=device)
|
| 62 |
+
model.tts_to_file(text, speaker_id, tmp_melo_path,speed=speed)
|
| 63 |
|
| 64 |
# Use speaker_wav as reference to extract style embedding
|
| 65 |
+
from openvoice import se_extractor
|
| 66 |
+
torch.hub.load('snakers4/silero-vad', 'silero_vad', force_reload=False)
|
| 67 |
+
ref_se, _ = se_extractor.get_se(speaker_wav, tone_color_converter, vad=True)
|
| 68 |
|
| 69 |
# Run the tone conversion
|
| 70 |
+
tone_color_converter.convert(
|
| 71 |
audio_src_path=tmp_melo_path,
|
| 72 |
src_se=ref_se,
|
| 73 |
tgt_se=ref_se,
|