From c87f5a2c46fea1bf2b0d00c8ab36286d07f19ac6 Mon Sep 17 00:00:00 2001
From: Henri Huotari <7366986+huotarih@users.noreply.github.com>
Date: Mon, 15 May 2023 11:05:30 +0300
Subject: [PATCH] upgrade rembg to be able to use isnet-general-use

---
 install.py                      | 2 +-
 scripts/postprocessing_rembg.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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