added missing function definition

README.md

@@ -23,9 +23,20 @@ https://huggingface.co/spaces/showlab/ShowUI
 import ast
 import torch
 from PIL import Image, ImageDraw
+import matplotlib.pyplot as plt
 from qwen_vl_utils import process_vision_info
 from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor
 
+
+def display(image, figsize=(10, 20)):
+    if isinstance(image, str):
+        image = Image.open(BytesIO(requests.get(image).content)) if image.startswith('http') else Image.open(image)
+    plt.figure(figsize=figsize)
+    plt.imshow(image)
+    plt.axis('off')
+    plt.show()
+
+
 def draw_point(image_input, point=None, radius=5):
     if isinstance(image_input, str):
         image = Image.open(BytesIO(requests.get(image_input).content)) if image_input.startswith('http') else Image.open(image_input)