Update README.md

README.md

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