Views
No views yet
transformers-compatible model (no dependency on the original training code)
and added a small classification head
(Linear(300→64) → BatchNorm1d → ReLU → Dropout(0.2) → Linear(64→50)).neerajaabhyankar/hindustani-raag-small
(50 raags), with an 85/15 held-out validation split for early stopping.1import datasets
2from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
3
4model = AutoModelForAudioClassification.from_pretrained(
5 "neerajaabhyankar/resnet-finetuned-1-hindustani-raag-small", trust_remote_code=True
6)
7feature_extractor = AutoFeatureExtractor.from_pretrained(
8 "neerajaabhyankar/resnet-finetuned-1-hindustani-raag-small", trust_remote_code=True
9)
10
11ds = datasets.load_dataset("neerajaabhyankar/hindustani-raag-small", split="test")
12audio = ds[0]["audio"]
13inputs = feature_extractor(audio["array"], audio["sampling_rate"])
14logits = model(inputs["input_values"].unsqueeze(0)).logits
15predicted_raag = model.config.id2label[logits.argmax(-1).item()]