add support for base64 str like "data:image/png;base64,xxxxxx...."

mrhan1993 · Dec 22, 2023 · 482cd35 · 482cd35
1 parent d10012b
commit 482cd35
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fooocusapi/img_utils.py b/fooocusapi/img_utils.py
@@ -36,6 +36,8 @@ def read_input_image(input_image: UploadFile) -> np.ndarray:
     return image
 
 def base64_to_stream(image: str) -> UploadFile:
+    if image.startswith('data:image'):
+        image = image.split(sep=',', maxsplit=1)[1]
     image_bytes = base64.b64decode(image)
     byte_stream = BytesIO()
     byte_stream.write(image_bytes)