pierre-loic commited on
Commit
197c21f
·
1 Parent(s): 741ce4a
Files changed (1) hide show
  1. tasks/audio.py +2 -1
tasks/audio.py CHANGED
@@ -55,7 +55,8 @@ async def evaluate_audio(request: AudioEvaluationRequest):
55
  #--------------------------------------------------------------------------------------------
56
  def extract_features(example, sampling_rate):
57
  audio_array = example['audio']['array']
58
- mfcc = librosa.feature.mfcc(y=audio_array, sr=sampling_rate, n_mfcc=5)
 
59
  return np.mean(mfcc, axis=1)
60
 
61
  def predict_new_audio(model, dataset, sampling_rate):
 
55
  #--------------------------------------------------------------------------------------------
56
  def extract_features(example, sampling_rate):
57
  audio_array = example['audio']['array']
58
+ # mfcc = librosa.feature.mfcc(y=audio_array, sr=sampling_rate, n_mfcc=5)
59
+ mfcc = librosa.feature.spectral_centroid(y=audio_array)
60
  return np.mean(mfcc, axis=1)
61
 
62
  def predict_new_audio(model, dataset, sampling_rate):