Commit
•
83ae482
1
Parent(s):
ba2084d
Update README.md
Browse files
README.md
CHANGED
|
@@ -40,6 +40,7 @@ Model reached 36.83% accuracy with training data and 37.02% with validation data
|
|
| 40 |
|
| 41 |
How to use it:
|
| 42 |
|
|
|
|
| 43 |
import tensorflow as tf
|
| 44 |
from transformers import WEIGHTS_NAME, CONFIG_NAME
|
| 45 |
from transformers import GPT2Config, TFGPT2LMHeadModel, GPT2Tokenizer
|
|
@@ -60,5 +61,6 @@ beam_outputs = model.generate(
|
|
| 60 |
no_repeat_ngram_size=2,
|
| 61 |
num_return_sequences=5
|
| 62 |
)
|
|
|
|
| 63 |
|
| 64 |
print(tokenizer.decode(beam_outputs[0]))
|
|
|
|
| 40 |
|
| 41 |
How to use it:
|
| 42 |
|
| 43 |
+
```python
|
| 44 |
import tensorflow as tf
|
| 45 |
from transformers import WEIGHTS_NAME, CONFIG_NAME
|
| 46 |
from transformers import GPT2Config, TFGPT2LMHeadModel, GPT2Tokenizer
|
|
|
|
| 61 |
no_repeat_ngram_size=2,
|
| 62 |
num_return_sequences=5
|
| 63 |
)
|
| 64 |
+
```
|
| 65 |
|
| 66 |
print(tokenizer.decode(beam_outputs[0]))
|