AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-18 01:25:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix gradio

Browse Source
This commit is contained in:
Daniel Gatis 2024-01-31 10:45:25 -03:00
parent 8eb37ef413
commit e60cf455b9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
rembg/commands/s_command.py
View File

@ -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,