Update app.py

app.py

@@ -151,11 +151,12 @@ def main():
                     with st.spinner("Adding Watermark..."):
                         #wav = my_read_file(wav,max_second_encode)
                         watermark = model.get_watermark(wav, default_sr)
+                        watermarked_audio = wav + watermark
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
-#                         st.write("Watermarked Audio:")
-#                         print("watermarked_audio:", watermarked_audio)
-#                         st.audio(watermarked_audio, format="audio/wav")
-#                         st.write("Time Cost: %d seconds" % encode_time_cost)
+                        st.write("Watermarked Audio:")
+                        print("watermarked_audio:", watermarked_audio)
+                        st.audio(watermarked_audio, format="audio/wav")
+                        st.write("Time Cost: %d seconds" % encode_time_cost)
 
 #                         # st.button("Add Watermark", disabled=False)
 #         elif action == "Decode Watermark":