diff --git a/demo/sequence-labeling/sequence_label.py b/demo/sequence-labeling/sequence_label.py index 2fe72b5b16d2864d78ed8b296ad6a410fb72f5b2..673f2ccf2f106a9f0fb488e05feb2aafdf1ab305 100644 --- a/demo/sequence-labeling/sequence_label.py +++ b/demo/sequence-labeling/sequence_label.py @@ -37,7 +37,7 @@ args = parser.parse_args() if __name__ == '__main__': # Load Paddlehub ERNIE pretrained model - module = hub.Module(name="ernie_v2_chinese_tiny") + module = hub.Module(name="ernie_tiny") inputs, outputs, program = module.context( trainable=True, max_seq_len=args.max_seq_len) if module.name.startswith("ernie_v2"): diff --git a/demo/text-classification/text_classifier.py b/demo/text-classification/text_classifier.py index abeabcdf6f055c0b9672b392ee338129751518d8..9dee887e959b312117820d6ac2c30d23bec7dbe7 100644 --- a/demo/text-classification/text_classifier.py +++ b/demo/text-classification/text_classifier.py @@ -42,7 +42,7 @@ if __name__ == '__main__': # Download dataset and use ClassifyReader to read dataset if args.dataset.lower() == "chnsenticorp": dataset = hub.dataset.ChnSentiCorp() - module = hub.Module(name="ernie_v2_chinese_tiny") + module = hub.Module(name="ernie_tiny") metrics_choices = ["acc"] elif args.dataset.lower() == "tnews": dataset = hub.dataset.TNews()