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,captions='Spectrogram of Mixture Speech (dB)')
+        st.image(spec_img)
         #showing mixed audio in streamlit 
         
         torchaudio.save("./waveform_mix.wav", waveform_mix, sr)
@@ -71,7 +71,7 @@ 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,captions='Spectrogram of Mixture Speech (dB)')
+        st.image(spec_clean_img)
         torchaudio.save("./waveform_souden.wav", waveform_souden, sr)
         st.audio("./waveform_souden.wav")