summaryrefslogtreecommitdiffstats
path: root/g4f/image.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/image.py')
-rw-r--r--g4f/image.py5
1 files changed, 2 insertions, 3 deletions
diff --git a/g4f/image.py b/g4f/image.py
index 4a97247e..01664f4e 100644
--- a/g4f/image.py
+++ b/g4f/image.py
@@ -64,7 +64,6 @@ def get_orientation(image: Image.Image) -> int:
def process_image(img: Image.Image, new_width: int, new_height: int) -> Image.Image:
orientation = get_orientation(img)
- new_img = Image.new("RGB", (new_width, new_height), color="#FFFFFF")
if orientation:
if orientation > 4:
img = img.transpose(Image.FLIP_LEFT_RIGHT)
@@ -74,8 +73,8 @@ def process_image(img: Image.Image, new_width: int, new_height: int) -> Image.Im
img = img.transpose(Image.ROTATE_270)
if orientation in [7, 8]:
img = img.transpose(Image.ROTATE_90)
- new_img.paste(img, (0, 0))
- return new_img
+ img.thumbnail((new_width, new_height))
+ return img
def to_base64(image: Image.Image, compression_rate: float) -> str:
output_buffer = BytesIO()