From 955ef9ec2a32301ce9ec84d0b23bf2ed79b4c857 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 20 Aug 2024 14:38:42 -0700 Subject: [PATCH] minor fix for ocr (#204) --- vision_agent/tools/tools.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 7837e207..a8339df8 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -312,6 +312,8 @@ def ocr(image: np.ndarray) -> List[Dict[str, Any]]: pil_image = Image.fromarray(image).convert("RGB") image_size = pil_image.size[::-1] + if image_size[0] < 1 and image_size[1] < 1: + return [] image_buffer = io.BytesIO() pil_image.save(image_buffer, format="PNG") buffer_bytes = image_buffer.getvalue()