Update app.py

app.py

@@ -138,7 +138,7 @@ if uploaded_file is not None:
     if st.button("Submit"):
         try:
             audio_to_spectrogram("uploaded_audio.wav")
-            st.image("spectrogram.png", caption="Mel Spectrogram of the uploaded audio file", use_column_width="auto", width=200)
+            st.image("spectrogram.png", caption="Mel Spectrogram of the uploaded audio file", use_container_width=True)
             speaker, probability, _ = classify_speaker("uploaded_audio.wav")
             gender, gen_probability = classify_gender("uploaded_audio.wav")
             
@@ -170,7 +170,7 @@ if recorded_audio:
 
     # Process the recorded audio
     audio_to_spectrogram("recorded_audio.wav")
-    st.image("spectrogram.png", caption="Mel Spectrogram of the recorded audio file", use_column_width="auto", width=200)
+    st.image("spectrogram.png", caption="Mel Spectrogram of the uploaded audio file", use_container_width=True)
 
     # Classify the speaker and gender
     speaker, probability, wav_file = classify_speaker("recorded_audio.wav")