Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ Since the generation relies on some randomness, we set a seed for reproducibilit
|
|
17 |
|
18 |
```python
|
19 |
>>> from transformers import pipeline, set_seed
|
20 |
-
>>> generator = pipeline('text-generation', model='nlp-waseda/comet-gpt2-small-japanese')
|
21 |
>>> set_seed(42)
|
22 |
>>> generator('X が 副業 を 始めるxEffect', max_length=30, num_return_sequences=5, do_sample=True)
|
23 |
|
|
|
17 |
|
18 |
```python
|
19 |
>>> from transformers import pipeline, set_seed
|
20 |
+
>>> generator = pipeline('text-generation', model='nlp-waseda/comet-v2-gpt2-small-japanese')
|
21 |
>>> set_seed(42)
|
22 |
>>> generator('X が 副業 を 始めるxEffect', max_length=30, num_return_sequences=5, do_sample=True)
|
23 |
|