diff --git a/demo/text_classification/predict.py b/demo/text_classification/predict.py
index 732df892e9519bdfc2eab5296477af41d22cf203..e53cf2b8712f1160abb99e985ca85fb5a4174127 100644
--- a/demo/text_classification/predict.py
+++ b/demo/text_classification/predict.py
@@ -40,7 +40,7 @@ args = parser.parse_args()
 
 if __name__ == '__main__':
     # Load Paddlehub ERNIE Tiny pretrained model
-    module = hub.Module(name="ernie_v2_eng_base")
+    module = hub.Module(name="ernie_tiny")
     inputs, outputs, program = module.context(
         trainable=True, max_seq_len=args.max_seq_len)
 
diff --git a/demo/text_classification/text_classifier.py b/demo/text_classification/text_classifier.py
index f987953a70d2ca337a803914a0c821a106f056c0..23746c03e2563ca2696ff0351cb93d73ae17de1f 100644
--- a/demo/text_classification/text_classifier.py
+++ b/demo/text_classification/text_classifier.py
@@ -36,7 +36,7 @@ args = parser.parse_args()
 if __name__ == '__main__':
 
     # Load Paddlehub ERNIE Tiny pretrained model
-    module = hub.Module(name="ernie_v2_eng_base")
+    module = hub.Module(name="ernie_tiny")
     inputs, outputs, program = module.context(
         trainable=True, max_seq_len=args.max_seq_len)