Skip to content

Commit 198a1ff

Browse files
committed
fix API returning extra stuff in base64 encoded iamges for #3972
1 parent fb39314 commit 198a1ff

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)