From b423f7b0900714742de43cb159cecbcc343aeeb8 Mon Sep 17 00:00:00 2001 From: pharmapsychotic <96542870+pharmapsychotic@users.noreply.github.com> Date: Fri, 4 Nov 2022 20:41:04 -0500 Subject: [PATCH] Fix for when uploaded image has alpha channel --- clip_interrogator.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clip_interrogator.ipynb b/clip_interrogator.ipynb index 9a53df4..39c8b4b 100644 --- a/clip_interrogator.ipynb +++ b/clip_interrogator.ipynb @@ -315,7 +315,7 @@ " print(uploader.value)\n", " print(uploader.value.items())\n", " for name, file_info in uploader.value.items():\n", - " image = Image.open(io.BytesIO(file_info['content']))\n", + " image = Image.open(io.BytesIO(file_info['content'])).convert('RGB')\n", " break\n", " else:\n", " if str(image_url).startswith('http://') or str(image_url).startswith('https://'):\n", @@ -2180,4 +2180,4 @@ }, "nbformat": 4, "nbformat_minor": 0 -} \ No newline at end of file +}