xtekky/gpt4free
1
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2024-11-26 21:21:21 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update image.py

Browse Source
This commit is contained in:
H Lohaus 2024-02-10 20:16:38 +01:00 committed by GitHub
parent ebcfa97c81
commit 4ba52b3695
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
g4f/image.py
View File

@ -163,10 +163,10 @@ def process_image(img: Image, new_width: int, new_height: int) -> Image:
# Resize image
img.thumbnail((new_width, new_height))
# Remove transparency
if img.mode != "RGB":
if img.mode == "RGBA":
img.load()
white = new_image('RGB', img.size, (255, 255, 255))
white.paste(img, mask=img.convert('RGBA').split()[-1])
white.paste(img, mask=img.split()[-1])
return white
return img