Update app.py

app.py

@@ -4,8 +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
+import scipy.io.wavfile as wavfile
+#import io
 
 # Load the pretrained model
 model = separator.from_hparams(source="speechbrain/sepformer-whamr-enhancement", savedir='pretrained_models/sepformer-whamr-enhancement')
@@ -28,13 +28,10 @@ def app():
         # Perform speech enhancement using the Sepformer model
         enhanced_speech = model.separate_file(path='uploaded_audio.wav')
         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
+        sample_rate = 8000
+        wavfile.write('enhanced_signal.wav', sample_rate, enhanced_signal_np)
+        enhanced_byte = 'enhanced_signal.wav'.read()
         st.audio(audio_bytes, format='audio/wav')
         st.audio(enhanced_byte, format='audio/wav')