SohomToom commited on
Commit
c1e585c
·
verified ·
1 Parent(s): 3d0281b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
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
- #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,
 
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,