Update app.py

app.py

@@ -220,7 +220,7 @@ def main():
 
                     elif file_extension == ".mp3":
                         torchaudio.save("output.mp3", squeeze, default_sr)
-                        st.audio("output.wav", format="audio/mpeg")
+                        st.audio("output.mp3", format="audio/mpeg")
 
                         # st.download_button(
                         #     label="Download Watermarked audio",