Audio Classification
ONNX
Russian
korallll commited on
Commit
108f355
·
verified ·
1 Parent(s): e742f64

Add contact: email + Telegram channel

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -28,4 +28,9 @@ model = WavLMForMusicDetection(batch_size=32, device='cuda')
28
  with safe_open('music_detection.safetensors', framework="pt") as f:
29
  model.load_state_dict({k: f.get_tensor(k) for k in f.keys()})
30
 
31
- probs = model.predict_proba(['audio1.mp3', 'audio2.wav']) # → tensor([0.88, 0.11])
 
 
 
 
 
 
28
  with safe_open('music_detection.safetensors', framework="pt") as f:
29
  model.load_state_dict({k: f.get_tensor(k) for k in f.keys()})
30
 
31
+ probs = model.predict_proba(['audio1.mp3', 'audio2.wav']) # → tensor([0.88, 0.11])
32
+
33
+ ## Contact
34
+
35
+ - Email: kborodin.research@gmail.com
36
+ - Telegram: [@korallll_ai](https://t.me/korallll_ai)