Update README.md

README.md

@@ -57,13 +57,14 @@ print(entities)
 Once you have the `entities`, you can use the following helper function to draw their bounding bboxes on the image:
 
 ```python
-import os
+import cv2
 import numpy as np
+import os
+import requests
 import torch
-from PIL import Image
 import torchvision.transforms as T
-import cv2
-import requests
+
+from PIL import Image
 
 
 def is_overlapping(rect1, rect2):
@@ -178,6 +179,10 @@ def draw_entity_boxes_on_image(image, entities, show=False, save_path=None):
     return new_image
 
 
+# (The same image from the previous code example)
+url = "https://huggingface.co/ydshieh/kosmos-2-patch14-224/resolve/main/snowman.jpg"
+image = Image.open(requests.get(url, stream=True).raw)
+
 # From the previous code example
 entities = [('a snowman', (12, 21), [(0.390625, 0.046875, 0.984375, 0.828125)]), ('a fire', (41, 47), [(0.171875, 0.015625, 0.484375, 0.890625)])]