Update app.py

app.py

@@ -51,11 +51,12 @@ def generate_audio():
         ).audios
 
         # Create an absolute file path in the "static/generated_audio" directory
+        output_audio = audio_output[0].T.float().cpu().numpy()
         output_filename = f"{prompt.replace(' ', '_')}_output.wav"
         output_file_path = os.path.join(PUBLIC_DIRECTORY, output_filename)
 
         # Write the generated audio to the file
-        sf.write(output_file_path, audio_output.T.cpu().numpy(), pipe.vae.sampling_rate)
+        sf.write(output_file_path, output_audio, pipe.vae.sampling_rate)
 
         # Return the absolute URL to the file
         download_url = f"/download/{output_filename}"