diff options
| author | Riccardo Giovanetti <29801031+Harvester62@users.noreply.github.com> | 2022-11-01 16:45:08 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-01 16:45:08 +0000 |
| commit | 5637ef33b9958abc6a1c2229ea342abf3b2c10fa (patch) | |
| tree | 9efd7b89bf8c198f8e716f3ed3d5ec99327fab6a /modules/api/api.py | |
| parent | a3eab2f71e53c56b2de767550250a93749c6630b (diff) | |
| parent | 198a1ffcfc963a3d74674fad560e87dbebf7949f (diff) | |
| download | stable-diffusion-webui-gfx803-5637ef33b9958abc6a1c2229ea342abf3b2c10fa.tar.gz stable-diffusion-webui-gfx803-5637ef33b9958abc6a1c2229ea342abf3b2c10fa.tar.bz2 stable-diffusion-webui-gfx803-5637ef33b9958abc6a1c2229ea342abf3b2c10fa.zip | |
Merge branch 'AUTOMATIC1111:master' into Italian
Diffstat (limited to 'modules/api/api.py')
| -rw-r--r-- | modules/api/api.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index e702c9c0..bb87d795 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -1,6 +1,8 @@ +import base64 +import io import time import uvicorn -from gradio.processing_utils import encode_pil_to_base64, decode_base64_to_file, decode_base64_to_image +from gradio.processing_utils import decode_base64_to_file, decode_base64_to_image from fastapi import APIRouter, Depends, HTTPException import modules.shared as shared from modules import devices @@ -29,6 +31,12 @@ def setUpscalers(req: dict): return reqDict +def encode_pil_to_base64(image): + buffer = io.BytesIO() + image.save(buffer, format="png") + return base64.b64encode(buffer.getvalue()) + + class Api: def __init__(self, app, queue_lock): self.router = APIRouter() |