Update app.py

app.py

@@ -74,6 +74,7 @@ def main():
 
         # Load the WAV file using torchaudio
         wav, sample_rate = torchaudio.load(wav_output_file)
+        wav.unsqueeze(0)
 
         # Play the audio file (WAV format)
         st.audio(wav_output_file, format="audio/wav")