Update app.py

app.py

@@ -47,7 +47,7 @@ def generate_audio():
             num_waveforms_per_prompt=1,
             generator=generator
         ).audios
-
+        '''
         # Convert to numpy and save to a WAV file
         output_audio = audio_output[0].T.float().cpu().numpy()
         output_filename = "output.wav"
@@ -56,6 +56,21 @@ def generate_audio():
 
         # Return the WAV file
         return jsonify({"file_path": output_path}), 200
+        '''
+        # Convert to numpy
+        output_audio = audio_output[0].T.float().cpu().numpy()
+
+        # Create a temporary file to store the output audio
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
+            temp_file_path = temp_file.name
+            # Save the audio to the temporary file
+            sf.write(temp_file_path, output_audio, pipe.vae.sampling_rate)
+            
+            # Return the file as an attachment
+            return send_file(temp_file_path, as_attachment=True, download_name="output.wav", mimetype="audio/wav")
+
+
+        
     except Exception as e:
         return jsonify({"error": str(e)}), 500