Update README.md

README.md

@@ -1 +1,9 @@
-prompt generation using gpt neo
+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)