Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import librosa
+import soundfile as sf
 import librosa.display
 from config import CONFIG
 import torch
@@ -109,9 +110,12 @@ if st.button('Conceal lossy audio!'):
         fig = visualize(target, lossy_input, output)
         st.pyplot(fig)
     st.success('Done!')
+    sf.write('target.wav', target, sr)
+    sf.write('lossy.wav', lossy_input, sr)
+    sf.write('enhanced.wav', output, sr)
     st.text('Original audio')
-    st.audio(target, sample_rate=sr)
+    st.audio('target.wav')
     st.text('Lossy audio')
-    st.audio(lossy_input, sample_rate=sr)
+    st.audio('lossy.wav')
     st.text('Enhanced audio')
-    st.audio(output, sample_rate=sr)
+    st.audio('enhanced.wav')