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

GPT2 trained to generate ЗНО (Ukrainian exam SAT type of thing) essays

Generated texts are not very cohesive yet but I'm working on it. The Hosted inference API outputs (on the right) are too short for some reason. Trying to fix it. Use the code from the example below. The model takes "ZNOTITLE: " inputs.

Example of usage:

from transformers import AlbertTokenizer, GPT2LMHeadModel
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)))