Update app.py

app.py

@@ -39,6 +39,7 @@ def predict_genre(audio):
     # audio = np.ndarray(audio)
     audio = audio.astype(np.float32)
     melspec = get_melspec_feature(audio, target_sr, frame_size, frame_shift_len, 128) 
+    melspec = melspec.reshape(-1, 431, 128, 1)
     prediction = model.predict(melspec)[0]
     return {genre_classes[i]: float(prediction[i]) for i in range(5)}