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)