Update app.py

app.py

@@ -4,6 +4,8 @@ import speechbrain as sb
 from speechbrain.dataio.dataio import read_audio
 from IPython.display import Audio
 from speechbrain.pretrained import SepformerSeparation as separator
+from scipy.io import wavfile
+import io
 
 # Load the pretrained model
 model = separator.from_hparams(source="speechbrain/sepformer-whamr-enhancement", savedir='pretrained_models/sepformer-whamr-enhancement')
@@ -25,11 +27,16 @@ def app():
 
         # Perform speech enhancement using the Sepformer model
         enhanced_speech = model.separate_file(path='uploaded_audio.wav')
-        enhanced_signal = Audio(enhanced_speech[:, :].detach().cpu().squeeze(),rate = 8000)
-
+        enhanced_signal = enhanced_speech[:, :].detach().cpu().squeeze()
+        
+        enhanced_bytes = io.BytesIO()
+        wavfile.write("enhanced_audio.wav", 8000, enhanced_signal.numpy())
+        with open("enhanced_audio.wav", "rb") as f:
+            enhanced_byte = f.read()
+        #st.audio(enhanced_byte, format='audio/wav')
         # Play the original and enhanced audio
         st.audio(audio_bytes, format='audio/wav')
-        st.audio(enhanced_signal, format='audio/wav')
+        st.audio(enhanced_byte, format='audio/wav')
 
 # Run the Streamlit app
 if __name__ == '__main__':