Update app.py

app.py

@@ -144,7 +144,7 @@ def main():
         action = st.selectbox("Select Action", ["Add Watermark", "Decode Watermark"])
 
         if action == "Add Watermark":
-            watermark_text = st.number_input("The watermark (0, 1 list of length-16):", value=st.session_state.def_value)
+            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:
@@ -168,7 +168,7 @@ def main():
                         # watermarked_audio = torchaudio.save("output.wav", squeeze, default_sr)
                         
                         #2nd Attempt
-                        watermarked_audio = model(wav, sample_rate=default_sr, message=watermark_text, alpha=1)
+                        watermarked_audio = model(wav, sample_rate=default_sr, alpha=1)
                         print(watermarked_audio.size())
                         size = watermarked_audio.size()
                         st.markdown(size)