gpt2-uk-zno-edition / README.md
kyryl0s's picture
Update README.md
d557a0b
|
raw
history blame
690 Bytes
metadata
license: afl-3.0
language: uk

GPT2 trained to generate ZNO (Ukrainian SAT) essays.

General info:

Generated texts are not very cohesive yet but I'm working on it.

Example of usage:

tokenizer = AlbertTokenizer.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
model = GPT2LMHeadModel.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
input_ids = tokenizer.encode("ZNOTITLE: Чи можна трахатися?", add_special_tokens=False, return_tensors='pt')

outputs = model.generate(
    input_ids,
    do_sample=True,
    num_return_sequences=1,
    max_length=250
)

for i, out in enumerate(outputs):
    print("{}: {}".format(i, tokenizer.decode(out)))