Kholykoko commited on
Commit
6fa1316
·
1 Parent(s): d92f370

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ def get_melspec_feature(X, target_sr, frame_size, frame_shift_len, n_mels):
35
 
36
 
37
  def predict_genre(audio):
 
38
  melspec = get_melspec_feature(audio, target_sr, frame_size, frame_shift_len, n_mels)
39
  prediction = model.predict(melspec)[0]
40
  return {genre_classes[i]: float(prediction[i]) for i in range(5)}
 
35
 
36
 
37
  def predict_genre(audio):
38
+ audio, _ = librosa.load(y = audio, sr = 22050)
39
  melspec = get_melspec_feature(audio, target_sr, frame_size, frame_shift_len, n_mels)
40
  prediction = model.predict(melspec)[0]
41
  return {genre_classes[i]: float(prediction[i]) for i in range(5)}