jimbozhang
commited on
Commit
·
828596e
1
Parent(s):
7166c88
Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ Augmentation and knowledge distillation (KD) are well-established techniques emp
|
|
21 |
|
22 |
### Model Sources
|
23 |
|
24 |
-
- **Repository:** https://github.com/RicherMans/CED
|
25 |
- **Paper:** [CED: Consistent ensemble distillation for audio tagging](https://arxiv.org/abs/2308.11957)
|
26 |
- **Demo:** https://huggingface.co/spaces/mispeech/ced-base
|
27 |
|
@@ -36,9 +36,9 @@ pip install -r requirements.txt
|
|
36 |
>>> from ced_model.feature_extraction_ced import CedFeatureExtractor
|
37 |
>>> from ced_model.modeling_ced import CedForAudioClassification
|
38 |
|
39 |
-
>>>
|
40 |
-
>>> feature_extractor = CedFeatureExtractor.from_pretrained(
|
41 |
-
>>> model = CedForAudioClassification.from_pretrained(
|
42 |
|
43 |
>>> import torchaudio
|
44 |
>>> audio, sampling_rate = torchaudio.load("resources/JeD5V5aaaoI_931_932.wav")
|
|
|
21 |
|
22 |
### Model Sources
|
23 |
|
24 |
+
- **Original Repository:** https://github.com/RicherMans/CED
|
25 |
- **Paper:** [CED: Consistent ensemble distillation for audio tagging](https://arxiv.org/abs/2308.11957)
|
26 |
- **Demo:** https://huggingface.co/spaces/mispeech/ced-base
|
27 |
|
|
|
36 |
>>> from ced_model.feature_extraction_ced import CedFeatureExtractor
|
37 |
>>> from ced_model.modeling_ced import CedForAudioClassification
|
38 |
|
39 |
+
>>> model_id = "mispeech/ced-tiny"
|
40 |
+
>>> feature_extractor = CedFeatureExtractor.from_pretrained(model_id)
|
41 |
+
>>> model = CedForAudioClassification.from_pretrained(model_id)
|
42 |
|
43 |
>>> import torchaudio
|
44 |
>>> audio, sampling_rate = torchaudio.load("resources/JeD5V5aaaoI_931_932.wav")
|