Skip to content

H
Hydra Node Http
Commit bbcb482d authored by kurumuz's avatar kurumuz
Browse files
Options

make PNG work, hopefully

parent 90b418ca
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 3 deletions
main.py
View file @ bbcb482d
...@@ -14,10 +14,12 @@ from typing import Union ...@@ -14,10 +14,12 @@ from typing import Union
import time import time
import gc import gc
import os import os
import io
import signal import signal
import simplejpeg import simplejpeg
import base64 import base64
import traceback import traceback
from PIL import Image
#Initialize model and config #Initialize model and config
model, config = init_config_model() model, config = init_config_model()
...@@ -99,7 +101,12 @@ def generate(request: GenerationRequest): ...@@ -99,7 +101,12 @@ def generate(request: GenerationRequest):
images_encoded = [] images_encoded = []
for x in range(len(images)): for x in range(len(images)):
image = simplejpeg.encode_jpeg(images[x], quality=95) #image = simplejpeg.encode_jpeg(images[x], quality=95)
image = Image.fromarray(images[x])
#save pillow image with bytesIO
output = io.BytesIO()
image.save(output, format='PNG')
image = output.getvalue()
#get base64 of image #get base64 of image
image = base64.b64encode(image).decode("ascii") image = base64.b64encode(image).decode("ascii")
images_encoded.append(image) images_encoded.append(image)
...@@ -150,7 +157,12 @@ def generate_advanced(request: GenerationRequest): ...@@ -150,7 +157,12 @@ def generate_advanced(request: GenerationRequest):
images_encoded = [] images_encoded = []
for x in range(len(images)): for x in range(len(images)):
image = simplejpeg.encode_jpeg(images[x], quality=95) #image = simplejpeg.encode_jpeg(images[x], quality=95)
image = Image.fromarray(images[x])
#save pillow image with bytesIO
output = io.BytesIO()
image.save(output, format='PNG')
image = output.getvalue()
#get base64 of image #get base64 of image
image = base64.b64encode(image).decode("ascii") image = base64.b64encode(image).decode("ascii")
images_encoded.append(image) images_encoded.append(image)
...@@ -215,7 +227,12 @@ def generate(request: GenerationRequest): ...@@ -215,7 +227,12 @@ def generate(request: GenerationRequest):
images = model.sample(request) images = model.sample(request)
images_encoded = [] images_encoded = []
for x in range(len(images)): for x in range(len(images)):
image = simplejpeg.encode_jpeg(images[x], quality=95) #image = simplejpeg.encode_jpeg(images[x], quality=95)
image = Image.fromarray(images[x])
#save pillow image with bytesIO
output = io.BytesIO()
image.save(output, format='PNG')
image = output.getvalue()
#get base64 of image #get base64 of image
image = base64.b64encode(image).decode("ascii") image = base64.b64encode(image).decode("ascii")
images_encoded.append(image) images_encoded.append(image)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment