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
+import io
 
 # pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
@@ -141,6 +142,13 @@ def main():
         st.markdown(wav)
         wav= wav.unsqueeze(0)
 
+        #2nd way
+        # Convert the tensor to a byte-like object in WAV format
+        with io.BytesIO() as buffer:
+        # Save the audio to the buffer using torchaudio
+            torchaudio.save(buffer, wav, default_sr, format="wav")
+        # Get the byte data from the buffer
+            wav = buffer.getvalue()
         # Play the audio file (WAV format)
         st.audio(wav, format="audio/wav")