Update app.py

app.py

@@ -153,14 +153,17 @@ def main():
             wav, sample_rate = torchaudio.load("test.wav")
             st.markdown("Before unsquueze wav")
             st.markdown(wav)
+
+            #Unsqueeze for line 176
+            wav= wav.unsqueeze(0)
                 
         elif file_extension == ".mp3":
             wav, sample_rate = librosa.load("test.mp3")
             st.markdown("Before unsqueeze mp3")
             st.markdown(wav)
         
-        #Unsqueeze for line 176
-        wav= wav.unsqueeze(0)
+        # #Unsqueeze for line 176
+        # wav= wav.unsqueeze(0)
 
         # #2nd way
         # # Convert the tensor to a byte-like object in WAV format