Update README.md

README.md

@@ -40,7 +40,6 @@ def download_image(url):
     image = image.convert("RGB")
     return image
 
-
 image         = download_image(URL)
 seed          = random.randint(0, 100000)
 generator     = torch.manual_seed(seed)
@@ -52,6 +51,6 @@ height        = int((height * factor) // 64) * 64
 image         = ImageOps.fit(image, (width, height), method=Image.Resampling.LANCZOS)
 
 prompt        = "Detect the person."
-images        = pipe(prompt, image=image, num_inference_steps=100, generator=generator).images
+images        = pipe(prompt, image=image, num_inference_steps=100, generator=generator).images[0]
 images[0]
 ```