Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: "cs"
|
| 3 |
+
tags:
|
| 4 |
+
- Czech
|
| 5 |
+
- KKY
|
| 6 |
+
- FAV
|
| 7 |
+
license: "cc-by-nc-sa-4.0"
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# gpt2-small-cs
|
| 11 |
+
Small version of GPT2 model (4 layers, 8 attention heads, hid. size 512) pretrained from 115 GB of cleaned Czech text (mainly from Common Crawl project).
|
| 12 |
+
|
| 13 |
+
## Usage
|
| 14 |
+
```python
|
| 15 |
+
from transformers import pipeline, set_seed
|
| 16 |
+
generator = pipeline('text-generation', model='fav-kky/gpt2-small-cs')
|
| 17 |
+
set_seed(42)
|
| 18 |
+
generator("Lidstvo bude brzy", max_length=30, num_return_sequences=2)
|
| 19 |
+
[{'generated_text': 'Lidstvo bude brzy seskupeno do osmi skupin, které budou rozděleny do čtyř skupin. V čele budou čtyři prezidenti, mezi nimi například prezidenti'},
|
| 20 |
+
{'generated_text': 'Lidstvo bude brzy potřebovat čas na budování infrastruktury, řekl na tiskové konferenci ředitel Národní síťové informační agentury Jiří Drahoš. "Napojení, které bude'}]
|
| 21 |
+
```
|