Update app.py

app.py

@@ -156,8 +156,8 @@ def main():
                         size = watermarked_audio.size()
                         st.markdown(size)
 
-                        print(watermarked_audio.shape())
-                        shape = watermarked_audio.size()
+                        print(watermarked_audio.squeeze())
+                        shape = watermarked_audio.squeeze()
                         st.markdown(shape)
                         
                         watermarked_audio = torchaudio.save("output.wav", watermarked_audio, default_sr)