flax-community
/

dansk-gpt-wiki

Text Generation Transformers PyTorch JAX TensorBoard Safetensors Danish gpt2 Inference Endpoints text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

dansk-gpt-wiki / evaluate.py

birgermoell's picture

Updated eval script

d1850c7 almost 3 years ago

raw history blame contribute delete

No virus

439 Bytes

	from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed

	tokenizer = GPT2Tokenizer.from_pretrained("flax-community/dansk-gpt-wiki")
	model = GPT2LMHeadModel.from_pretrained("flax-community/dansk-gpt-wiki")


	generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
	set_seed(42)
	result = generator("Jeg elsker livet", max_length=150, num_return_sequences=5)
	print(result)