switin06 commited on
Commit
856b229
·
verified ·
1 Parent(s): 8288bf1

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -1
inference.py CHANGED
@@ -231,12 +231,13 @@ def text_to_speech(text, output_path, speed=1.3):
231
 
232
  # Load multilingual multi-speaker TTS model
233
  tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=True, gpu=False)
 
234
 
235
  # Choose a male speaker (e.g., "male-en-1")
236
  male_speaker = "male-en-1"
237
 
238
  # Generate TTS to file with speaker
239
- tts.tts_to_file(text=text, speaker=male_speaker, file_path=raw_path)
240
 
241
  # Speed up using ffmpeg
242
  os.system(f"ffmpeg -y -i {raw_path} -filter:a atempo={speed} {output_path}")
 
231
 
232
  # Load multilingual multi-speaker TTS model
233
  tts = TTS(model_name="tts_models/multilingual/multi-dataset/your_tts", progress_bar=True, gpu=False)
234
+ language="en";
235
 
236
  # Choose a male speaker (e.g., "male-en-1")
237
  male_speaker = "male-en-1"
238
 
239
  # Generate TTS to file with speaker
240
+ tts.tts_to_file(text=text, speaker=male_speaker,language=language,file_path=raw_path)
241
 
242
  # Speed up using ffmpeg
243
  os.system(f"ffmpeg -y -i {raw_path} -filter:a atempo={speed} {output_path}")