Update README.md
Browse files
README.md
CHANGED
@@ -16,6 +16,9 @@ KoGPT2 on Huggingface Transformers
|
|
16 |
```python
|
17 |
from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
|
18 |
|
|
|
|
|
|
|
19 |
input_ids = tokenizer.encode("์๋
", add_special_tokens=False, return_tensors="pt")
|
20 |
output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=100, num_return_sequences=3)
|
21 |
for generated_sequence in output_sequences:
|
|
|
16 |
```python
|
17 |
from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
|
18 |
|
19 |
+
model = GPT2LMHeadModel.from_pretrained("taeminlee/kogpt2")
|
20 |
+
tokenizer = PreTrainedTokenizerFast.from_pretrained("taeminlee/kogpt2")
|
21 |
+
|
22 |
input_ids = tokenizer.encode("์๋
", add_special_tokens=False, return_tensors="pt")
|
23 |
output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=100, num_return_sequences=3)
|
24 |
for generated_sequence in output_sequences:
|