fix(api): match inpaint source and mask formats

ssube · Nov 26, 2023 · 7ca5dc1 · 7ca5dc1
1 parent 93e3125
commit 7ca5dc1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py
@@ -252,7 +252,7 @@ def inpaint(server: ServerContext, pool: DevicePoolExecutor):
     if mask_file is None:
         return error_reply("mask image is required")
 
-    source = Image.open(BytesIO(source_file.read())).convert("RGB")
+    source = Image.open(BytesIO(source_file.read())).convert("RGBA")
     size = Size(source.width, source.height)
 
     mask_top_layer = Image.open(BytesIO(mask_file.read())).convert("RGBA")