We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fb39314 commit 198a1ffCopy full SHA for 198a1ff
modules/api/api.py
@@ -1,6 +1,8 @@
1
+import base64
2
+import io
3
import time
4
import uvicorn
-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
6
from fastapi import APIRouter, Depends, HTTPException
7
import modules.shared as shared
8
from modules import devices
@@ -29,6 +31,12 @@ def setUpscalers(req: dict):
29
31
return reqDict
30
32
33
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
40
class Api:
41
def __init__(self, app, queue_lock):
42
self.router = APIRouter()
0 commit comments