Browse Source

fix gradio

Daniel Gatis 1 year ago
parent
commit
e60cf455b9
1 changed files with 2 additions and 1 deletions
  1. 2 1
      rembg/commands/s_command.py

+ 2 - 1
rembg/commands/s_command.py

@@ -261,8 +261,9 @@ def s_command(port: int, host: str, log_level: str, threads: int) -> None:
         return await asyncify(im_without_bg)(file, commons)  # type: ignore
 
     def gr_app(app):
-        def inference(input_path, model, cmd_args):
+        def inference(input_path, model, *args):
             output_path = "output.png"
+            a, af, ab, ae, om, ppm, cmd_args = args
 
             kwargs = {
                 "alpha_matting": a,