app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import os
 import cv2
 
-def inference(file, af, mask):
+def inference(file, mask, af):
     im = cv2.imread(file, cv2.IMREAD_COLOR)
     cv2.imwrite(os.path.join("input.png"), im)
 
@@ -28,11 +28,23 @@ article = "<p style='text-align: center;'><a href='https://github.com/danielgati
 
 gr.Interface(
     inference, 
-    [gr.inputs.Image(type="filepath", label="Input"), gr.Slider(10, 25, value=10, label="Alpha matting"), gr.Radio(choices = ["Alpha matting", "Mask only"], value = "Alpha matting")], 
+    [
+        gr.inputs.Image(type="filepath", label="Input"),
+        gr.inputs.Slider(10, 25, default=10, label="Alpha matting"), 
+        gr.inputs.Radio(
+            [
+                "Default", 
+                "Mask only"
+            ], 
+            type="value",
+            default="Alpha matting",
+            label="Choices"
+        )
+    ], 
     gr.outputs.Image(type="file", label="Output"),
     title=title,
     description=description,
     article=article,
-    examples=[['lion.png']],
+    examples=[["lion.png", 10, "Default"], ["girl.jpg", 10, "Default"]],
     enable_queue=True
     ).launch()

rembg/_version.py

@@ -24,8 +24,8 @@ def get_keywords():
     # each be defined on a line of their own. _version.py will just call
     # get_keywords().
     git_refnames = " (HEAD -> main)"
-    git_full = "edc9fe27dff030cf6c2f29ef9a66c32d6e3f4658"
-    git_date = "2022-11-28 08:14:19 -0300"
+    git_full = "d62227d5866e2178e88f06074917484a4424082e"
+    git_date = "2022-12-10 11:51:49 -0300"
     keywords = {"refnames": git_refnames, "full": git_full, "date": git_date}
     return keywords
 

rembg/session_base.py

@@ -18,7 +18,7 @@ class BaseSession:
         std: Tuple[float, float, float],
         size: Tuple[int, int],
     ) -> Dict[str, np.ndarray]:
-        im = img.convert("RGB").resize(size, Image.Resampling.LANCZOS)
+        im = img.convert("RGB").resize(size, Image.LANCZOS)
 
         im_ary = np.array(im)
         im_ary = im_ary / np.max(im_ary)

rembg/session_simple.py

@@ -25,6 +25,6 @@ class SimpleSession(BaseSession):
         pred = np.squeeze(pred)
 
         mask = Image.fromarray((pred * 255).astype("uint8"), mode="L")
-        mask = mask.resize(img.size, Image.Resampling.LANCZOS)
+        mask = mask.resize(img.size, Image.LANCZOS)
 
         return [mask]