Update README.md

README.md

@@ -68,13 +68,13 @@ pipe.to("cuda")
 # text-to-image
 prompt = "impressionist painting of an astronaut in a jungle"
 ret = pipe(prompt)
-ret.images[0].save("astronaut.png")
+ret.image.save("astronaut.png")
 
 # image editing
 image = Image.open(requests.get('https://github.com/baaivision/Emu/Emu2/examples/dog.jpg?raw=true',stream=True).raw).convert('RGB')
 prompt = [image, "wearing a red hat on the beach."]
 ret = pipe(prompt)
-ret.images[0].save("dog_hat_beach.png")
+ret.image.save("dog_hat_beach.png")
 
 # grounding generation
 def draw_box(left, top, right, bottom):
@@ -110,7 +110,7 @@ prompt = [
 	dog3,
 ]
 ret = pipe(prompt)
-ret.images[0].save("three_dogs.png")
+ret.image.save("three_dogs.png")
 
 # Autoencoding
 # to enable the autoencoding mode, you can only input exactly one image as prompt