Kholykoko commited on
Commit
5d3642a
·
1 Parent(s): 604d33b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -37,6 +37,7 @@ def predict_genre(audio):
37
  sr, audio = audio
38
  # print(audio)
39
  # audio = np.ndarray(audio)
 
40
  melspec = get_melspec_feature(audio, target_sr, frame_size, frame_shift_len, 128)
41
  prediction = model.predict(melspec)[0]
42
  return {genre_classes[i]: float(prediction[i]) for i in range(5)}
 
37
  sr, audio = audio
38
  # print(audio)
39
  # audio = np.ndarray(audio)
40
+ audio = audio.astype(numpy.float32)
41
  melspec = get_melspec_feature(audio, target_sr, frame_size, frame_shift_len, 128)
42
  prediction = model.predict(melspec)[0]
43
  return {genre_classes[i]: float(prediction[i]) for i in range(5)}