Update app.py

app.py

@@ -47,7 +47,7 @@ def ui():
         waveform_clean = waveform_clean.to(torch.double)
         stft_clean = stft(waveform_clean)
         st.text("Your uploaded audio")
-        st.audio(waveform_clean)
+        st.audio(audio_file)
         #creating a mixture of our audio file and the noise file 
         waveform_mix = generate_mixture(waveform_clean, waveform_noise, target_snr)
         #making the files into torch double format 
@@ -58,7 +58,9 @@ def ui():
         spec_img=plot_spectrogram(stft_mix)
         st.image(spec_img,captions='Spectrogram of Mixture Speech (dB)')
         #showing mixed audio in streamlit 
-        st.audio(waveform_mix)
+        
+        torchaudio.save("./waveform_mix.wav", waveform_mix, sr)
+        st.audio("./waveform_mix.wav")
         #getting the irms 
         irm_speech, irm_noise = get_irms(stft_clean, stft_noise)
         #getting the psd speech
@@ -70,7 +72,8 @@ def ui():
         spec_clean_img=plot_spectrogram(stft_souden)
         waveform_souden = waveform_souden.reshape(1, -1)
         st.image(spec_clean_img,captions='Spectrogram of Mixture Speech (dB)')
-        st.audio(waveform_souden)
+        torchaudio.save("./waveform_souden.wav", waveform_souden, sr)
+        st.audio("./waveform_souden.wav")
 
 if __name__=="__main__": 
     ui()