BoldActionMan commited on
Commit
7a2edb2
·
verified ·
1 Parent(s): 030d75b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -9
app.py CHANGED
@@ -160,18 +160,22 @@ def process_video(video_file, youtube_url, language_choice):
160
 
161
  segment_files = []
162
  subtitle_entries = []
163
- for segment in translation_segments:
164
- segment_file, start, end, translated_text = generate_segment_audio(segment, speaker_id)
165
-
 
 
 
 
166
  # Run the tone color converter
167
  encode_message = "@MyShell"
168
  tone_color_converter.convert(
169
- audio_src_path=segment_file,
170
- src_se=source_se,
171
- tgt_se=target_se,
172
- output_path=segment_file,
173
- message=encode_message)
174
-
175
  segment_files.append((segment_file, start, end, translated_text))
176
 
177
  # Combine the audio segments
 
160
 
161
  segment_files = []
162
  subtitle_entries = []
163
+ for i in range(0, len(translation_segments), batch_size):
164
+ batch = translation_segments[i:i + batch_size]
165
+ with ThreadPoolExecutor(max_workers=5) as executor:
166
+ batch_segment_files = list(executor.map(generate_segment_audio_batch, [batch] * len(speaker_ids), [speaker_id] * len(speaker_ids)))
167
+ batch_segment_files = [item for sublist in batch_segment_files for item in sublist] # Flatten the list
168
+
169
+ for segment_file, start, end, translated_text in batch_segment_files:
170
  # Run the tone color converter
171
  encode_message = "@MyShell"
172
  tone_color_converter.convert(
173
+ audio_src_path=segment_file,
174
+ src_se=source_se,
175
+ tgt_se=target_se,
176
+ output_path=segment_file,
177
+ message=encode_message
178
+ )
179
  segment_files.append((segment_file, start, end, translated_text))
180
 
181
  # Combine the audio segments