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 +}