Update app.py

app.py

@@ -157,11 +157,11 @@ def main():
                         st.markdown(size)
 
                         print(watermarked_audio.squeeze())
-                        squeeze = watermarked_audio.squeeze()
+                        squeeze = watermarked_audio.squeeze(1)
                         shape = squeeze.size()
                         st.markdown(shape)
                         
-                        watermarked_audio = torchaudio.save("output.wav", watermarked_audio, default_sr)
+                        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)