xynull's picture
Add application file
cbe8be1
raw
history blame contribute delete
205 Bytes
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)