Update README.md

README.md

@@ -4,10 +4,20 @@ language:
 - zh
 pipeline_tag: text-generation
 widget:
-- text: "生命、宇宙以及一切的终极答案是"
+- text: "中国的首都是"
 
 ---
 
 The model is based on [bigscience/bloom-560m](https://huggingface.co/bigscience/bloom-560m). 
 
-To reduce GPU memory usage, we pruned its vocabulary from 250880 to 42437 with Chinese corpus. So the total parameter is 389m now.
+To reduce GPU memory usage, we pruned its vocabulary from 250880 to 42437 with Chinese corpus. So the total parameter is 389m now.
+
+# How to use
+```python
+from transformers import BloomTokenizerFast, BloomForCausalLM
+
+tokenizer = BloomTokenizerFast.from_pretrained('Langboat/bloom-389m-zh')
+model = BloomForCausalLM.from_pretrained('Langboat/bloom-389m-zh')
+
+print(tokenizer.batch_decode(model.generate(tokenizer.encode('中国的首都是', return_tensors='pt'))))
+```