Update app.py

app.py

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