Update app.py

app.py

@@ -10,7 +10,6 @@ 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")
 
@@ -153,12 +152,12 @@ def main():
                         #wav = my_read_file(wav,max_second_encode)
                         watermark = model.get_watermark(wav, default_sr)
                         watermarked_audio = wav + watermark
+                        watermarked_audio = soundfile.write("output.wav", watermarked_audio, default_sr)
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
                         st.write("Watermarked Audio:")
                         st.markdown(watermarked_audio)
                         print("watermarked_audio:", watermarked_audio)
-                        #st.audio(watermarked_audio, format="audio/wav")
-                        play_audio(watermarked_audio)
+                        st.audio(watermarked_audio, format="audio/wav")
                         #st.write("Time Cost: %d seconds" % encode_time_cost)
 
 #                         # st.button("Add Watermark", disabled=False)