Skip to content

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

make PNG work, hopefully

parent 90b418ca
Show 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
import time
import gc
import os
import io
import signal
import simplejpeg
import base64
import traceback
from PIL import Image
#Initialize model and config
model, config = init_config_model()
......@@ -99,7 +101,12 @@ def generate(request: GenerationRequest):
images_encoded = []
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
image = base64.b64encode(image).decode("ascii")
images_encoded.append(image)
......@@ -150,7 +157,12 @@ def generate_advanced(request: GenerationRequest):
images_encoded = []
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
image = base64.b64encode(image).decode("ascii")
images_encoded.append(image)
......@@ -215,7 +227,12 @@ def generate(request: GenerationRequest):
images = model.sample(request)
images_encoded = []
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
image = base64.b64encode(image).decode("ascii")
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