From e60cf455b9277b85267758f031fec6c3f6afe3a8 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Wed, 31 Jan 2024 10:45:25 -0300 Subject: [PATCH] fix gradio --- rembg/commands/s_command.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rembg/commands/s_command.py b/rembg/commands/s_command.py index 04661e2..a6cc276 100644 --- a/rembg/commands/s_command.py +++ b/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,