Update app.py

app.py

@@ -148,10 +148,16 @@ def main():
         # mp3_audio.export(wav_output_file, format="wav")
 
         # Load the WAV file using torchaudio
-        wav, sample_rate = torchaudio.load("test.wav")
-        st.markdown("Before unsquuezewav")
-        st.markdown(wav)
-
+        if file_extension in [".wav", ".flac"]: 
+            wav, sample_rate = torchaudio.load("test.wav")
+            st.markdown("Before unsquueze wav")
+            st.markdown(wav)
+                
+        elif file_extension == ".mp3":
+            wav, sample_rate = torchaudio.load("test.mp3")
+            st.markdown("Before unsqueeze mp3")
+            st.markdown(wav)
+        
         #Unsqueeze for line 176
         wav= wav.unsqueeze(0)
 
@@ -168,7 +174,7 @@ def main():
         # wav, sample_rate = torchaudio.load(audio_file, format="mp3/wav")
         st.markdown("SR")
         st.markdown(sample_rate)
-        st.markdown("after unsqueeze wav")
+        st.markdown("after unsqueeze wav or mp3")
         st.markdown(wav)
         # 展示文件到页面上
         # st.audio(tmp_input_audio_file, format="audio/wav")