Update app.py

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import soundfile
 from wavmark.utils import file_reader
 
+
 # pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
 # st.title("Hot Dog? Or Not?")
@@ -65,10 +66,10 @@ def main():
         if action == "Add Watermark":
             watermark_text = st.text_input("The watermark (0, 1 list of length-16):", value=st.session_state.def_value)
             add_watermark_button = st.button("Add Watermark", key="add_watermark_btn")
-#             if add_watermark_button:  # 点击按钮后执行的
-#                 if audio_file and watermark_text:
-#                     with st.spinner("Adding Watermark..."):
-#                         watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
+            if add_watermark_button:  # 点击按钮后执行的
+                if audio_file and watermark_text:
+                    with st.spinner("Adding 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")