RoGPT2-medium / README.md
MihaiAlexandru1606's picture
init
b0c1531

Model card for RoGPT2-medium


language: - ro

RoGPT2: Romanian GPT2 for text generation

All models are available:

How to use


from transformers import AutoTokenizer, TFAutoModelWithLMHead

tokenizer = AutoTokenizer.from_pretrained('readerbench/RoGPT2-medium')
model = TFAutoModelWithLMHead.from_pretrained('readerbench/RoGPT2-medium')
inputs = tokenizer.encode("Este o zi de vara", return_tensors='tf')
text = model.generate(inputs, max_length=1024,  no_repeat_ngram_size=2)
print(tokenizer.decode(text[0]))