소스 검색

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,