mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
Use image.thumbnail
This commit is contained in:
parent
104212529f
commit
0abff499e2
@ -89,15 +89,7 @@ def get_SD_pictures(description):
|
||||
image.save(output_file.as_posix())
|
||||
pic_id += 1
|
||||
# lower the resolution of received images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history
|
||||
width, height = image.size
|
||||
if (width > 300):
|
||||
height = int(height * (300 / width))
|
||||
width = 300
|
||||
elif (height > 300):
|
||||
width = int(width * (300 / height))
|
||||
height = 300
|
||||
newsize = (width, height)
|
||||
image = image.resize(newsize, Image.LANCZOS)
|
||||
image.thumbnail((300, 300))
|
||||
buffered = io.BytesIO()
|
||||
image.save(buffered, format="JPEG")
|
||||
buffered.seek(0)
|
||||
|
@ -26,15 +26,7 @@ def caption_image(raw_image):
|
||||
def generate_chat_picture(picture, name1, name2):
|
||||
text = f'*{name1} sends {name2} a picture that contains the following: "{caption_image(picture)}"*'
|
||||
# lower the resolution of sent images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history
|
||||
width, height = picture.size
|
||||
if (width > 300):
|
||||
height = int(height * (300 / width))
|
||||
width = 300
|
||||
elif (height > 300):
|
||||
width = int(width * (300 / height))
|
||||
height = 300
|
||||
newsize = (width, height)
|
||||
picture = picture.resize(newsize, Image.LANCZOS)
|
||||
image.thumbnail((300, 300))
|
||||
buffer = BytesIO()
|
||||
picture.save(buffer, format="JPEG")
|
||||
img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
|
||||
|
Loading…
Reference in New Issue
Block a user