Update app.py

app.py

@@ -163,11 +163,12 @@ def main():
                     if file_extension in [".wav", ".flac"]:
                         torchaudio.save("output.wav", squeeze, default_sr, bits_per_sample=16)
                         watermarked_wav = torchaudio.save("output.wav", squeeze, default_sr, bits_per_sample=16)
-                        binary_data = watermarked_wav.getvalue()
+
                         st.audio("output.wav", format="audio/wav")
 
                         with open("output.wav", "wb") as file:
-                            data = file.read()
+                            file.write(watermarked_wav.getbuffer())
+                            binary_data = file.read()
                             btn = st.download_button(
                                 label="Download watermarked audio",
                                 data=binary_data,
@@ -183,7 +184,8 @@ def main():
                         st.audio("output.mp3", format="audio/mpeg")
                         
                         with open("output.mp3", "wb") as file:
-                            data = file.read()
+                            file.write(watermarked_wav.getbuffer())
+                            binary_data = file.read()
                             st.download_button(
                                 label="Download watermarked audio",
                                 data=binary_data,