Update app.py

app.py

@@ -164,26 +164,29 @@ def main():
                         torchaudio.save("output.wav", squeeze, default_sr, bits_per_sample=16)
                         watermarked_wav = torchaudio.save("output.wav", squeeze, default_sr, bits_per_sample=16)
                         st.audio("output.wav", format="audio/wav")
-                        
-                        st.download_button(
-                            label="Download watermarked audio",
-                            data=watermarked_wav,
-                            file_name="output.wav",
-                            mime="audio/wav",
-                        )
+
+
+                        with open("output.wav", "wb") as file:
+                            btn = st.download_button(
+                                label="Download watermarked audio",
+                                data=file,
+                                file_name="output.wav",
+                                mime="audio/wav",
+                            )
 
 
                     elif file_extension == ".mp3":
                         torchaudio.save("output.mp3", squeeze, default_sr)
                         watermarked_mp3 = torchaudio.save("output.mp3", squeeze, default_sr)
                         st.audio("output.mp3", format="audio/mpeg")
-
-                        st.download_button(
-                            label="Download watermarked audio",
-                            data=watermarked_mp3,
-                            file_name="output.mp3",
-                            mime="audio/mpeg",
-                        )
+                        
+                        with open("output.mp3", "wb") as file:
+                            st.download_button(
+                                label="Download watermarked audio",
+                                data=file,
+                                file_name="output.mp3",
+                                mime="audio/mpeg",
+                            )
 
                         # st.download_button(
                         #     label="Download Watermarked audio",