Update app.py

app.py

@@ -30,8 +30,10 @@ def app():
         enhanced_signal = enhanced_speech[:, :].detach().cpu().squeeze()
         # Play the original and enhanced audio
         sample_rate = 8000
-        wavfile.write('enhanced_signal.wav', sample_rate, enhanced_signal)
-        enhanced_byte = 'enhanced_signal.wav'.read()
+        enhanced_signal = enhanced_signal.reshape((1,enhanced_signal.shape[0]))
+        torchaudio.save('enhanced_sound.wav', enhanced_signal, sample_rate)        enhanced_byte = 'enhanced_signal.wav'.read()
+        with open('amazing_sound.wav', 'rb') as f:
+            enhanced_byte = f.read()
         st.audio(audio_bytes, format='audio/wav')
         st.audio(enhanced_byte, format='audio/wav')