Merge branch 'main' into dev

mrhan1993 · Dec 22, 2023 · 5efa725 · 5efa725
2 parents 1431422 + 49639d6
commit 5efa725
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)