Commit
•
c71020c
1
Parent(s):
4297407
Update README.md
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ When the parameter skip_special_tokens is False:
|
|
35 |
|
36 |
```python
|
37 |
>>> from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
|
38 |
-
>>> tokenizer = BertTokenizer.from_pretrained("
|
39 |
>>> model = GPT2LMHeadModel.from_pretrained("JeffreyLau/SikuGPT2-poem")
|
40 |
>>> text_generator = TextGenerationPipeline(model, tokenizer)
|
41 |
>>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=100, do_sample=True)
|
|
|
35 |
|
36 |
```python
|
37 |
>>> from transformers import BertTokenizer, GPT2LMHeadModel,TextGenerationPipeline
|
38 |
+
>>> tokenizer = BertTokenizer.from_pretrained("JeffreyLau/SikuGPT2-poem")
|
39 |
>>> model = GPT2LMHeadModel.from_pretrained("JeffreyLau/SikuGPT2-poem")
|
40 |
>>> text_generator = TextGenerationPipeline(model, tokenizer)
|
41 |
>>> text_generator("[CLS]梅 山 如 积 翠 ,", max_length=100, do_sample=True)
|