Update app.py

app.py

@@ -164,8 +164,10 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
             
             # Ensure both segments have the same duration
             max_length = max(len(mixed_audio), len(overlay_audio))
-            mixed_audio = mixed_audio.pad_to_width(max_length)
-            overlay_audio = overlay_audio.pad_to_width(max_length)
+            if len(mixed_audio) < max_length:
+                mixed_audio += AudioSegment.silent(duration=max_length - len(mixed_audio))
+            if len(overlay_audio) < max_length:
+                overlay_audio += AudioSegment.silent(duration=max_length - len(overlay_audio))
             
             # Overlay the audio
             mixed_audio = mixed_audio.overlay(overlay_audio)