Update app.py

app.py

@@ -152,23 +152,8 @@ def main():
                         #wav = my_read_file(wav,max_second_encode)
                         
                         #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, alpha=1)
+                        watermark = model.get_watermark(wav, default_sr)
+                        watermarked_audio = wav + watermark
                         print(watermarked_audio.size())
                         size = watermarked_audio.size()
                         st.markdown(size)
@@ -179,13 +164,29 @@ def main():
                         st.markdown(shape)
 
                         st.markdown(squeeze)
-                        # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
-                        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")
+                        
+                        #2nd Attempt
+                        # watermarked_audio = model(wav, sample_rate=default_sr, alpha=1)
+                        # 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, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
+                        # 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)
 
 #                         # st.button("Add Watermark", disabled=False)