Update README.md

README.md

@@ -36,10 +36,9 @@ The Transformer-XL model was pre-trained on the Wudao Corpus (with 280G samples)
  ```
 
 ```python 
-from fengshen.models.transfo_xl_paraphrase import (
-    TransfoXLModel,
-    TransfoXLTokenizer
-)
+from fengshen.models.transfo_xl_paraphrase import TransfoXLModel
+from transformers import T5Tokenizer as TransfoXLTokenizer
+
 
 model = TransfoXLModel.from_pretrained('IDEA-CCNL/Randeng-TransformerXL-1.1B-Paraphrasing-Chinese')
 tokenizer = TransfoXLTokenizer.from_pretrained('IDEA-CCNL/Randeng-TransformerXL-1.1B-Paraphrasing-Chinese',
@@ -52,7 +51,7 @@ tokenizer = TransfoXLTokenizer.from_pretrained('IDEA-CCNL/Randeng-TransformerXL-
 ```python 
 from fengshen.models.transfo_xl_paraphrase import paraphrase_generatete
 input_text = "年轻教师选择农村学校，还是县城学校？"                                                 
-res = paraphrase_generate(model, tokenizer, input_text, device=1)                                             
+res = paraphrase_generate(model, tokenizer, input_text, device=0)                                             
 print(res)
 #
 ```