fix --input file.jpg

2024-01-31 22:10:13 +08:00
parent 1d2d39dfa0
commit 44ee0769dd
4 changed files with 35 additions and 31 deletions
--- a/iopaint/helper.py
+++ b/iopaint/helper.py
@@ -304,7 +304,9 @@ def get_image_ext(img_bytes):
 def decode_base64_to_image(
    encoding: str, gray=False
 ) -> Tuple[np.array, Optional[np.array], Dict]:
-    if encoding.startswith("data:image/"):
+    if encoding.startswith("data:image/") or encoding.startswith(
+        "data:application/octet-stream;base64,"
+    ):
        encoding = encoding.split(";")[1].split(",")[1]
    image = Image.open(io.BytesIO(base64.b64decode(encoding)))