Update app.py

app.py

@@ -10,6 +10,7 @@ from wavmark.utils import file_reader
 from audioseal import AudioSeal
 import torchaudio
 from pydub import AudioSegment
+from notebook import play_audio, plot_waveform_and_specgram
 
 # pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
@@ -157,6 +158,7 @@ def main():
                         st.markdown(watermarked_audio)
                         print("watermarked_audio:", watermarked_audio)
                         #st.audio(watermarked_audio, format="audio/wav")
+                        play_audio(watermarked_audio)
                         #st.write("Time Cost: %d seconds" % encode_time_cost)
 
 #                         # st.button("Add Watermark", disabled=False)