fix rembg: rename arg image to input_image.

lankii · Jul 15, 2024 · 643ffc8 · 643ffc8
1 parent 7eed6f1
commit 643ffc8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/webuiapi/webuiapi.py b/webuiapi/webuiapi.py
@@ -1270,7 +1270,7 @@ def __init__(self, webuiapi):
 
     def rembg(
         self,
-        image: str = "", #image string (?)
+        input_image: str = "", #image string (?)
         model: str = 'u2net',  #[None, 'u2net', 'u2netp', 'u2net_human_seg', 'u2net_cloth_seg','silueta','isnet-general-use','isnet-anime']
         return_mask: bool = False,
         alpha_matting: bool = False,
@@ -1280,7 +1280,7 @@ def rembg(
     ):
 
         payload = {
-            "image": b64_img(image),
+            "input_image": b64_img(input_image),
             "model": model,
             "return_mask": return_mask,
             "alpha_matting":  alpha_matting,