diff --git a/install.py b/install.py index 5c13936..8ec2c14 100644 --- a/install.py +++ b/install.py @@ -1,7 +1,7 @@ import launch if not launch.is_installed("rembg"): - launch.run_pip("install rembg==2.0.30 --no-deps", "rembg") + launch.run_pip("install rembg==2.0.36 --no-deps", "rembg") for dep in ['onnxruntime', 'pymatting', 'pooch']: if not launch.is_installed(dep): diff --git a/scripts/postprocessing_rembg.py b/scripts/postprocessing_rembg.py index da80e86..3e1417f 100644 --- a/scripts/postprocessing_rembg.py +++ b/scripts/postprocessing_rembg.py @@ -6,6 +6,7 @@ import rembg models = [ "None", + "isnet-general-use", "u2net", "u2netp", "u2net_human_seg",