Fix formatting

README.md

@@ -38,7 +38,7 @@ Check out [**Fast Anime PromptGen**](https://huggingface.co/FredZhang7/anime-any
 See the Prompt Generator tab of [Paint Journey Demo](https://huggingface.co/spaces/FredZhang7/paint-journey-demo).
 
 
-### Contrastive Search
+## Contrastive Search
 
 ```bash
 pip install --upgrade transformers
@@ -68,11 +68,11 @@ for i in range(len(output)):
     print('\033[92m' + tokenizer.decode(output[i], skip_special_tokens=True) + '\033[0m\n')
 ```
 
-Example output:
+Example of generated prompts:
 ![constrastive search](./constrastive_search.png)
 
 
-### Greedy Search
+## Greedy Search
 
 Replace `output` with
 
@@ -80,5 +80,5 @@ Replace `output` with
 output = model.generate(input_ids, do_sample=True, temperature=temperature, top_k=top_k, max_length=max_length, num_return_sequences=num_return_sequences, early_stopping=True)
 ```
 
-Example output:
+Example of generated prompts:
 ![greedy search](./greedy_search.png)