Update app.py

app.py

@@ -152,7 +152,7 @@ def main():
                         #wav = my_read_file(wav,max_second_encode)
                         watermark = model.get_watermark(wav, default_sr)
                         watermarked_audio = wav + watermark
-                        watermarked_audio = soundfile.write("output.wav", watermarked_audio, default_sr)
+                        watermarked_audio = torchaudio.save("output.wav", watermarked_audio, default_sr)
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
                         st.write("Watermarked Audio:")
                         st.markdown(watermarked_audio)