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

Fix process_image to handle transparency correctly

Browse Source
This commit is contained in:
Rafael Fonseca 2024-02-10 11:42:53 -03:00
parent 2250e37fec
commit b2f1c73019
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
g4f/image.py
View File

@ -166,7 +166,7 @@ def process_image(img: Image, new_width: int, new_height: int) -> Image:
if img.mode != "RGB": if img.mode != "RGB":
img.load() img.load()
white = new_image('RGB', img.size, (255, 255, 255)) white = new_image('RGB', img.size, (255, 255, 255))
white.paste(img, mask=img.split()[3]) white.paste(img, mask=img.convert('RGBA').split()[-1])
return white return white
return img return img