Update README.md

README.md

@@ -15,16 +15,16 @@ pip install transformers
 ```
 
 ```python
-from transformers import CPMAntTokenizer, CPMAntModel
+from transformers import CPMAntTokenizer, CPMAntForCausalLM
 
-texts = ["今天天气真好！"]
-model = CPMAntModel.from_pretrained("openbmb/cpm-ant-10b")
+texts = "今天天气真好！"
+model = CPMAntForCausalLM.from_pretrained("openbmb/cpm-ant-10b")
 tokenizer = CPMAntTokenizer.from_pretrained("openbmb/cpm-ant-10b")
-input_ids = tokenizer.get_model_input(texts)
-logits, hidden = model(**input_ids)
+input_ids = tokenizer(texts, return_tensors="pt")
+outputs = model.generate(**input_ids)
+output_texts = tokenizer.batch_decode(outputs)
 
-print(logits.shape)
-print(hidden.shape)
+print(output_texts)
 ```
 ## pipeline
 ```python