diff --git a/demos/audio_tagging/README.md b/demos/audio_tagging/README.md index e8f609886..c2b77aa5d 100644 --- a/demos/audio_tagging/README.md +++ b/demos/audio_tagging/README.md @@ -26,7 +26,7 @@ wget https://paddlespeech.bj.bcebos.com/PaddleAudio/cat.wav https://paddlespeech ``` Command usage: - `input`(required): Audio file to tag. - - `model`: Model type of tagging task. Default: `panns_cnn10`. + - `model`: Model type of tagging task. Default: `panns_cnn14`. - `config`: Config of tagging task. Use pretrained model when it is None. Default: `None`. - `ckpt_path`: Model checkpoint. Use pretrained model when it is None. Default: `None`. - `label_file`: Label file of tagging task. Use audioset labels when it is None. Default: `None`. diff --git a/demos/audio_tagging/tag.py b/demos/audio_tagging/tag.py index cda3c5ad4..e056122f4 100644 --- a/demos/audio_tagging/tag.py +++ b/demos/audio_tagging/tag.py @@ -27,7 +27,7 @@ args = parser.parse_args() if __name__ == '__main__': cls_executor = CLSExecutor() result = cls_executor( - model_type='panns_cnn10', + model_type='panns_cnn14', cfg_path=None, # Set `cfg_path` and `ckpt_path` to None to use pretrained model. label_file=None, ckpt_path=None,