Update README.md

README.md

@@ -34,16 +34,21 @@ pipeline = AutoPipelineForText2Image.from_pretrained(repo_id,
                                                      torch_dtype=bnb_4bit_compute_dtype
 )
 pipeline.enable_model_cpu_offload()
+```
 
+```python
 image = pipeline(
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k", num_inference_steps=28, guidance_scale=3.5, height=768
 ).images[0]
 image
+```
 
+![Astronaut in a jungle, cold color palette, muted colors, detailed, 8k](astro_1.png)
+
+```python
 image = pipeline(
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k, yarn art style", num_inference_steps=28, guidance_scale=3.5, height=768
 ).images[0]
 image
 ```
-
-![A cat wearing a wizard hat, photorealistic](astro_1.png)
+![Astronaut in a jungle, cold color palette, muted colors, detailed, 8k, yarn art style](astro_yarn.png)