MoHamdyy commited on
Commit
3c32f04
·
1 Parent(s): 8fbabff

Fix syntax error in TTS stage and complete pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -497,7 +497,7 @@ def full_speech_translation_pipeline(audio_input_path: str):
497
  try:
498
  print("TTS: Synthesizing English speech...")
499
  sequence = text_to_seq(english_translation).unsqueeze(0).to(DEVICE)
500
- generated_mel, _ = TTS_MODEL.inference(sequence, max_length=hp.max_mel_time-50, stop_token_threshold=0.2, with_tqdm=False)
501
 
502
  print(f"TTS: Generated mel shape: {generated_mel.shape if generated_mel is not None else 'None'}")
503
  if generated_mel is not None and generated_mel.numel() > 0:
 
497
  try:
498
  print("TTS: Synthesizing English speech...")
499
  sequence = text_to_seq(english_translation).unsqueeze(0).to(DEVICE)
500
+ generated_mel, _ = TTS_MODEL.inference(sequence, max_length=hp.max_mel_time-50, stop_token_threshold=0.5, with_tqdm=False)
501
 
502
  print(f"TTS: Generated mel shape: {generated_mel.shape if generated_mel is not None else 'None'}")
503
  if generated_mel is not None and generated_mel.numel() > 0: