modified: app.py

app.py

@@ -535,7 +535,11 @@ def generate_music(
         final_audio, vocal_audio, instrumental_audio = None, None, None
         try:
             final_audio_data, sr = sf.read(final_mix_path)
+            vocoder_audio_data, sr = sf.read(vocoder_mix)
+            recons_audio, sr = sf.read(recons_mix)
             final_audio = (sr, (final_audio_data * 32767).astype(np.int16))
+            vocal_audio = (sr, (vocoder_audio_data*32767).astype(np.int16))
+            instrumental_audio = (sr, (recons_audio*32767).astype(np.int16))
         except Exception as e:
             print("Final mix read error:", e)
         return final_audio, vocal_audio, instrumental_audio