Update README.md

README.md

@@ -40,15 +40,18 @@ def download_image(url):
     image = image.convert("RGB")
     return image
 
-image = download_image(URL)
+
+image         = download_image(URL)
+seed          = random.randint(0, 100000)
+generator     = torch.manual_seed(seed)
 width, height = image.size
-factor = 512 / max(width, height)
-factor = math.ceil(min(width, height) * factor / 64) * 64 / min(width, height)
-width = int((width * factor) // 64) * 64
-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).images
+factor        = 512 / max(width, height)
+factor        = math.ceil(min(width, height) * factor / 64) * 64 / min(width, height)
+width         = int((width * factor) // 64) * 64
+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[0]
 ```