File size: 264 Bytes
878041e |
1 2 3 4 5 6 7 8 |
from transformers import pipeline
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
prompt = 'The thing I hate most' # can be anything
res = generator(prompt, max_length=50, do_sample=True, temperature=0.7)
print(res[0]['generated_text']) |