Update app.py

app.py

@@ -159,6 +159,8 @@ def main():
                 
         elif file_extension == ".mp3":
             wav, sample_rate = librosa.load("test.mp3")
+            wav = torch.tensor(wav).float() / 32768.0
+            wav = audio.unsqueeze(0).unsqueeze(0)
             st.markdown("Before unsqueeze mp3")
             st.markdown(wav)