Update app.py

app.py

@@ -183,6 +183,8 @@ def main():
                         st.write("Watermarked Audio:")
                         st.markdown(watermarked_audio)
                         print("watermarked_audio:", watermarked_audio)
+
+                        watermarked_audio = torchaudio.save("output.wav", squeeze, default_sr)
                         st.audio(watermarked_audio, format="audio/wav")
                         #st.write("Time Cost: %d seconds" % encode_time_cost)