fix the photo upload issue

Signed-off-by: Stephanie <[email protected]>
containers · Jul 24, 2024 · 860393e · 860393e
1 parent 9465129
commit 860393e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/recipes/computer_vision/object_detection/app/object_detection_client.py b/recipes/computer_vision/object_detection/app/object_detection_client.py
@@ -23,6 +23,8 @@
     window.image(img, use_column_width=True)  
     # convert PIL image into bytes for post request 
     bytes_io = io.BytesIO() 
+    if img.mode in ("RGBA", "P"): 
+        img = img.convert("RGB")
     img.save(bytes_io, "JPEG")
     img_bytes = bytes_io.getvalue()
     b64_image = base64.b64encode(img_bytes).decode('utf-8')