thugCodeNinja commited on
Commit
a6fde96
1 Parent(s): 5de0202

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -30,6 +30,8 @@ def classify_audio(audio):
30
  # Convert the audio data to NumPy array
31
  rate, ar = audio
32
  arone = ar.astype(np.float32)
 
 
33
  mel_spectrogram = librosa.feature.melspectrogram(y=arone, sr=SAMPLE_RATE, n_mels=N_MELS)
34
  mel_spectrogram = librosa.power_to_db(mel_spectrogram, ref=np.max)
35
 
 
30
  # Convert the audio data to NumPy array
31
  rate, ar = audio
32
  arone = ar.astype(np.float32)
33
+ if arone.ndim > 1:
34
+ arone = arone.flatten()
35
  mel_spectrogram = librosa.feature.melspectrogram(y=arone, sr=SAMPLE_RATE, n_mels=N_MELS)
36
  mel_spectrogram = librosa.power_to_db(mel_spectrogram, ref=np.max)
37