Update app.py

app.py

@@ -56,7 +56,7 @@ def ui():
         stft_mix = stft(waveform_mix)
         #plotting the spectogram 
         spec_img=plot_spectrogram(stft_mix)
-        st.image(spec_img)
+        # st.image(spec_img)
         #showing mixed audio in streamlit 
         
         torchaudio.save("./waveform_mix.wav", waveform_mix, sr)
@@ -71,24 +71,10 @@ def ui():
         #plotting the cleaned audio and hearing it
         spec_clean_img=plot_spectrogram(stft_souden)
         waveform_souden = waveform_souden.reshape(1, -1)
-        st.image(spec_clean_img)
+        # st.image(spec_clean_img)
         torchaudio.save("./waveform_souden.wav", waveform_souden, sr)
         st.audio("./waveform_souden.wav")
 
 if __name__=="__main__": 
     ui()
 
-
-
-    
-
-
-
-        
-
-
-         
-    
-
-
-