prompt generation using gpt neo
Use it in transformers :-

from transformers import AutoTokenizer, AutoModelForCausalLM
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("DrDrek/prompt_generation")
model = AutoModelForCausalLM.from_pretrained("DrDrek/prompt_generation")
generator = pipeline('text-generation', model=model,tokenizer=tokenizer)
generator("white horse in clouds,", do_sample=True, min_length=20)