Update app.py

app.py

@@ -47,9 +47,10 @@ def generate_audio():
         ).audios
 
         # Create a temporary file to store the generated audio
+        output_audio = audio_output[0].T.float().cpu().numpy()
         with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
             # Write the audio to the temporary file
-            sf.write(tmpfile.name, audio_output.T.cpu().numpy(), pipe.vae.sampling_rate)
+            sf.write(tmpfile.name, output_audio, pipe.vae.sampling_rate)
 
             # Store the temporary file path
             temp_file_path = tmpfile.name