Update app.py

app.py

@@ -163,14 +163,14 @@ 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()
                             btn = st.download_button(
                                 label="Download watermarked audio",
-                                data=data,
+                                data=binary_data,
                                 file_name="output.wav",
                                 mime="audio/wav",
                             )
@@ -179,13 +179,14 @@ def main():
                     elif file_extension == ".mp3":
                         torchaudio.save("output.mp3", squeeze, default_sr)
                         watermarked_mp3 = torchaudio.save("output.mp3", squeeze, default_sr)
+                        binary_data = watermarked_mp3.getvalue()
                         st.audio("output.mp3", format="audio/mpeg")
                         
                         with open("output.mp3", "wb") as file:
                             data = file.read()
                             st.download_button(
                                 label="Download watermarked audio",
-                                data=data,
+                                data=binary_data,
                                 file_name="output.mp3",
                                 mime="audio/mpeg",
                             )