Skip to content

Commit 5637ef3

Browse files
authored
Merge branch 'AUTOMATIC1111:master' into Italian
2 parents a3eab2f + 198a1ff commit 5637ef3

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

modules/api/api.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
1+
import base64
2+
import io
13
import time
24
import uvicorn
3-
from gradio.processing_utils import encode_pil_to_base64, decode_base64_to_file, decode_base64_to_image
5+
from gradio.processing_utils import decode_base64_to_file, decode_base64_to_image
46
from fastapi import APIRouter, Depends, HTTPException
57
import modules.shared as shared
68
from modules import devices
@@ -29,6 +31,12 @@ def setUpscalers(req: dict):
2931
return reqDict
3032

3133

34+
def encode_pil_to_base64(image):
35+
buffer = io.BytesIO()
36+
image.save(buffer, format="png")
37+
return base64.b64encode(buffer.getvalue())
38+
39+
3240
class Api:
3341
def __init__(self, app, queue_lock):
3442
self.router = APIRouter()

0 commit comments

Comments
 (0)