Update README.md

README.md

@@ -66,6 +66,9 @@ pip install -v -e .
 [![Huggingface VisualCloze](https://img.shields.io/static/v1?label=Demo&message=Huggingface%20Gradio&color=orange)](https://huggingface.co/spaces/VisualCloze/VisualCloze)
 
 Example with Depth-to-Image:
+
+<img src="./visualcloze_diffusers_example_depthtoimage.jpg" width="30%" height="20%" alt="Example with Depth-to-Image"/>
+
 ```python
 import torch
 from diffusers import VisualClozePipeline
@@ -111,7 +114,7 @@ image_result = pipe(
     num_inference_steps=30,
     max_sequence_length=512,
     generator=torch.Generator("cpu").manual_seed(0)
-).images[0]
+).images[0][0]
 
 # Save the resulting image
 image_result.save("visualcloze.png")
@@ -161,7 +164,7 @@ image_result = pipe(
     num_inference_steps=30,
     max_sequence_length=512,
     generator=torch.Generator("cpu").manual_seed(0)
-).images[0]
+).images[0][0]
 
 # Save the resulting image
 image_result.save("visualcloze.png")