Update app.py

app.py

@@ -153,9 +153,10 @@ def main():
                         watermark = model.get_watermark(wav, default_sr)
                         watermarked_audio = wav + watermark
                         print(watermarked_audio.size())
+                        print(watermarked_audio.shape()))
                         shape = watermarked_audio.size()
                         st.markdown(shape)
-
+                        
                         watermarked_audio = torchaudio.save("output.wav", watermarked_audio, default_sr)
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
                         st.write("Watermarked Audio:")