Update app.py

app.py

@@ -46,9 +46,10 @@ def generate_segment_audio(text, lang, speaker_url, pipe):
  audio_np = audio_data_resampled.cpu().numpy()
  return audio_np
 
-# Function to concatenate audio segments
 def concatenate_audio_segments(segments):
- concatenated_audio = np.concatenate(segments, axis=0)
+ max_len = max([seg.shape[0] for seg in segments])
+ padded_segments = [np.pad(seg, (0, max_len - seg.shape[0]), 'constant') for seg in segments]
+ concatenated_audio = np.concatenate(padded_segments, axis=0)
  concatenated_audio = concatenated_audio / np.max(np.abs(concatenated_audio))
  return np.asarray(concatenated_audio, dtype=np.float32)