File size: 205 Bytes
d4aaa06
 
 
cbe8be1
 
 
1
2
3
4
5
6
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='gpt2')
set_seed(42)
r = generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)

print(r)