Kholykoko commited on
Commit
37ea4c7
·
1 Parent(s): 5d3642a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def predict_genre(audio):
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)}
 
37
  sr, audio = audio
38
  # print(audio)
39
  # audio = np.ndarray(audio)
40
+ audio = audio.astype(np.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)}