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,
|