2nd way of watermarking audio

app.py

@@ -150,8 +150,25 @@ def main():
                 if audio_file and watermark_text:
                     with st.spinner("Adding Watermark..."):
                         #wav = my_read_file(wav,max_second_encode)
-                        watermark = model.get_watermark(wav, default_sr)
-                        watermarked_audio = wav + watermark
+                        
+                        #1st attempt
+                        # watermark = model.get_watermark(wav, default_sr)
+                        # watermarked_audio = wav + watermark
+                        # print(watermarked_audio.size())
+                        # size = watermarked_audio.size()
+                        # st.markdown(size)
+
+                        # print(watermarked_audio.squeeze())
+                        # squeeze = watermarked_audio.squeeze(1)
+                        # shape = squeeze.size()
+                        # st.markdown(shape)
+
+                        # st.markdown(squeeze)
+                        
+                        # watermarked_audio = torchaudio.save("output.wav", squeeze, default_sr)
+                        
+                        #2nd Attempt
+                        watermarked_audio = model(wav, sample_rate=default_sr, message=watermark_text, alpha=1)
                         print(watermarked_audio.size())
                         size = watermarked_audio.size()
                         st.markdown(size)
@@ -162,8 +179,6 @@ def main():
                         st.markdown(shape)
 
                         st.markdown(squeeze)
-                        
-                        watermarked_audio = torchaudio.save("output.wav", squeeze, default_sr)
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
                         st.write("Watermarked Audio:")
                         st.markdown(watermarked_audio)