Downloads · 30 days
0
music-genre-classifier
mpuneeth17/music-genre-classifier
music-genre-classifier is a machine learning model from mpuneeth17. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
A EFFICIENTNET model trained on the GTZAN dataset that classifies music audio into 10 genres:
Access
Public
Updated Jul 8, 2026
Repo size
50 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth50 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Jul 8, 2026
- Updated
- Jul 8, 2026
- SHA
- 5c15a061
Datasets
Model card
From the Hugging Face model README
🎵 Music Genre Classifier
A EFFICIENTNET model trained on the GTZAN dataset that classifies music audio into 10 genres:
Blues · Classical · Country · Disco · Hip-Hop · Jazz · Metal · Pop · Reggae · Rock
Usage
import torch
import librosa, numpy as np
from torchvision import transforms
from PIL import Image
# Load model
model = ... # see app/app.py for architecture
ckpt = torch.hub.load_state_dict_from_url(
"https://huggingface.co/mpuneeth17/music-genre-classifier/resolve/main/best_model.pth",
map_location="cpu"
)
model.load_state_dict(ckpt["model"])
model.eval()
Demo
👉 Try it live: Hugging Face Space
Training
- Architecture: efficientnet
- Dataset: GTZAN (10 genres × 100 songs × 10 segments = 10,000 samples)
- Input: 128×128 Mel Spectrogram images
- Optimizer: AdamW + CosineAnnealingLR
- Accuracy: ~88-92%
- License
- mit
- Languages
- en
- Created
- Jul 8, 2026
- Updated
- Jul 8, 2026