Update README.md

README.md

@@ -13,13 +13,13 @@ Gpt-2 for short story generation with genres.
 ## Model description
 
 Gpt-2 model fine-tuned on sample of BookCorpus dataset for short story generation, allows for the following genres (tokens to use as input under parenthesis):
-- Romance ("<romance>")
-- Adventure ("<adventure>")
-- Mystery & detective ("<mystery-&-detective>")
-- Fantasy ("<fantasy>")
-- Humor & comedy ("<humor-&-comedy>")
-- Paranormal ("<paranormal>")
-- Science fiction ("<science-fiction>")
+- Romance (romance)
+- Adventure (adventure)
+- Mystery & detective (mystery-&-detective)
+- Fantasy (fantasy)
+- Humor & comedy (humor-&-comedy)
+- Paranormal (paranormal)
+- Science fiction (science-fiction)
 
 Heavily inspired by https://huggingface.co/pranavpsv
 ## Intended uses & limitations
@@ -34,7 +34,7 @@ This can be used for text generation.
 >>> model = GPT2LMHeadModel.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> story_generator = TextGenerationPipeline(model=model, tokenizer=tokenizer)
-# 
+# Input should be of format "<BOS> <Genre token> Optional starter text"
 >>> input_prompt = "<BOS> <adventure> He was trapped in the pirate's house"
 >>> story = story_generator(input_prompt, max_length=80, do_sample=True,
                repetition_penalty=1.5, temperature=1.2,