|
|
--- |
|
|
license: cc-by-nc-4.0 |
|
|
language: |
|
|
- ru |
|
|
base_model: |
|
|
- microsoft/wavlm-base-plus |
|
|
pipeline_tag: audio-classification |
|
|
--- |
|
|
# Music Detection with WavLM |
|
|
|
|
|
Detects if audio contains music. |
|
|
**EER: 2.5–3%** | Based on `microsoft/wavlm-base-plus` |
|
|
*the best threshold value* `0.2442` |
|
|
--- |
|
|
## Quick Start |
|
|
``` |
|
|
git clone https://huggingface.co/MTUCI/MusicDetection |
|
|
cd MusicDetection |
|
|
pip install -r requirements.txt |
|
|
``` |
|
|
## Usage |
|
|
|
|
|
```python |
|
|
from model import WavLMForMusicDetection |
|
|
from safetensors import safe_open |
|
|
|
|
|
model = WavLMForMusicDetection(batch_size=32, device='cuda') |
|
|
with safe_open('music_detection.safetensors', framework="pt") as f: |
|
|
model.load_state_dict({k: f.get_tensor(k) for k in f.keys()}) |
|
|
|
|
|
probs = model.predict_proba(['audio1.mp3', 'audio2.wav']) # → tensor([0.88, 0.11]) |
