| 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) |