Update README.md

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).
 
 
-### PyTorch
+### Contrastive Search
 
 ```bash
 pip install --upgrade transformers
@@ -69,4 +69,16 @@ for i in range(len(output)):
 ```
 
 Example output:
-![constrastive search](./constrastive_search.png)
+![constrastive search](./constrastive_search.png)
+
+
+### Greedy Search
+
+Replace `output` with
+
+```python
+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:
+![greedy search](./greedy_search.png)