Update README.md

README.md

@@ -1,3 +1,11 @@
+---
+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.  
@@ -20,6 +28,4 @@ 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])
-
-
+probs = model.predict_proba(['audio1.mp3', 'audio2.wav'])  # → tensor([0.88, 0.11])