kyryl0s commited on
Commit
d557a0b
1 Parent(s): d997c1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -1,3 +1,26 @@
1
  ---
2
  license: afl-3.0
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: afl-3.0
3
+ language: uk
4
  ---
5
+ ## GPT2 trained to generate ZNO (Ukrainian SAT) essays.
6
+
7
+ ### General info:
8
+ Generated texts are not very cohesive yet but I'm working on it.
9
+
10
+ ### Example of usage:
11
+ ```python
12
+ tokenizer = AlbertTokenizer.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
13
+ model = GPT2LMHeadModel.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
14
+ input_ids = tokenizer.encode("ZNOTITLE: Чи можна трахатися?", add_special_tokens=False, return_tensors='pt')
15
+
16
+ outputs = model.generate(
17
+ input_ids,
18
+ do_sample=True,
19
+ num_return_sequences=1,
20
+ max_length=250
21
+ )
22
+
23
+ for i, out in enumerate(outputs):
24
+ print("{}: {}".format(i, tokenizer.decode(out)))
25
+
26
+ ```