BoldActionMan commited on
Commit
f86b927
·
verified ·
1 Parent(s): d90eff5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -154,11 +154,14 @@ def process_video(video_file, language_choice):
154
  model = TTS(language=language, device=device)
155
  speaker_ids = model.hps.data.spk2id
156
 
157
- def generate_segment_audio(segment, speaker_id):
158
- start, end, translated_text = segment
159
- segment_path = os.path.join(output_dir, f'segment_{start}_{end}.wav')
160
- model.tts_to_file(translated_text, speaker_id, segment_path, speed=speed)
161
- return segment_path, start, end, translated_text
 
 
 
162
 
163
  for speaker_key in speaker_ids.keys():
164
  speaker_id = speaker_ids[speaker_key]
 
154
  model = TTS(language=language, device=device)
155
  speaker_ids = model.hps.data.spk2id
156
 
157
+ def generate_segment_audio_batch(translation_batch, speaker_id):
158
+ segment_files = []
159
+ for segment in translation_batch:
160
+ start, end, translated_text = segment
161
+ segment_path = os.path.join(output_dir, f'segment_{start}_{end}.wav')
162
+ model.tts_to_file(translated_text, speaker_id, segment_path, speed=speed)
163
+ segment_files.append((segment_path, start, end, translated_text))
164
+ return segment_files
165
 
166
  for speaker_key in speaker_ids.keys():
167
  speaker_id = speaker_ids[speaker_key]