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

GPT2 trained to generate ЗНО (Ukrainian exam SAT type of thing) 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)))