Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ Use the code from the example below. The model takes "ZNOTITLE: your essay title
|
|
14 |
from transformers import AlbertTokenizer, GPT2LMHeadModel
|
15 |
tokenizer = AlbertTokenizer.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
|
16 |
model = GPT2LMHeadModel.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
|
17 |
-
input_ids = tokenizer.encode("ZNOTITLE:
|
18 |
|
19 |
outputs = model.generate(
|
20 |
input_ids,
|
|
|
14 |
from transformers import AlbertTokenizer, GPT2LMHeadModel
|
15 |
tokenizer = AlbertTokenizer.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
|
16 |
model = GPT2LMHeadModel.from_pretrained("kyryl0s/gpt2-uk-zno-edition")
|
17 |
+
input_ids = tokenizer.encode("ZNOTITLE: За яку працю треба більше поважати людину - за фізичну чи інтелектуальну?", add_special_tokens=False, return_tensors='pt')
|
18 |
|
19 |
outputs = model.generate(
|
20 |
input_ids,
|