Merge pull request #105 from mrhan1993/main

add support for base64 str like "data:image/png;base64,xxxxxx...." in v2 endpoint
mrhan1993 · Dec 22, 2023 · 49639d6 · 49639d6
2 parents d10012b + 482cd35
commit 49639d6
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)