Update README.md

README.md

@@ -29,11 +29,13 @@ datasets:
 
 # 四、 使用
 ```python
-import torch
-from transformers.models.electra import ElectraModel, ElectraTokenizer
-tokenizer = ElectraTokenizer.from_pretrained("junnyu/electra_small_generator")
-model = ElectraModel.from_pretrained("junnyu/electra_small_generator")
-inputs = tokenizer("Beijing is the capital of [MASK].", return_tensors="pt")
-with torch.no_grad():
-    outputs = model(**inputs)
+from transformers import pipeline
+fill_mask = pipeline(
+	"fill-mask",
+	model="junnyu/electra_small_generator",
+	tokenizer="junnyu/electra_small_generator"
+)
+print(
+	fill_mask("HuggingFace is creating a [MASK] that the community uses to solve NLP tasks.")
+)
 ```