File size: 541 Bytes
e42b364 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import joblib
import numpy as np
from scripts.utilities import predict_audio_class,convert_mp3_to_wav
from tensorflow.keras.models import load_model
model = load_model('models/audio_extractor_model.keras')
converter = joblib.load('models/preprocessing/label_encoder.joblib')
music_file = input("Enter path to music file")
convert_mp3_to_wav(music_file,'temp/music.wav')
predaud = predict_audio_class(model, 'temp/music.wav')
predicted_label = converter.inverse_transform([np.argmax(predaud)])
print("Predicted Genre : ", predicted_label) |