瀏覽代碼

fix gradio

Daniel Gatis 1 年之前
父節點
當前提交
e60cf455b9
共有 1 個文件被更改,包括 2 次插入1 次删除
  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,