shibing624 commited on
Commit
f352b2b
1 Parent(s): 32144f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -39,7 +39,9 @@ pip install -U textgen
39
  ```
40
 
41
  ```python
 
42
  from textgen import BartSeq2SeqModel
 
43
  tokenizer = BertTokenizerFast.from_pretrained('shibing624/bart4csc-base-chinese')
44
  model = BartSeq2SeqModel(
45
  encoder_type='bart',
 
39
  ```
40
 
41
  ```python
42
+ from transformers import BertTokenizerFast
43
  from textgen import BartSeq2SeqModel
44
+
45
  tokenizer = BertTokenizerFast.from_pretrained('shibing624/bart4csc-base-chinese')
46
  model = BartSeq2SeqModel(
47
  encoder_type='bart',