Merge branch 'feature/add-inpaint-mask-generation'

# Conflicts:
#	webui.py
This commit is contained in:
Manuel Schmid
2024-01-25 10:48:40 +01:00
4 changed files with 49 additions and 23 deletions
+3 -2
View File
@@ -1,7 +1,7 @@
from rembg import remove, new_session
def generate_mask_from_image(image, mask_model):
def generate_mask_from_image(image, mask_model, extras):
if image is None:
return
@@ -11,5 +11,6 @@ def generate_mask_from_image(image, mask_model):
return remove(
image,
session=new_session(mask_model),
only_mask=True
only_mask=True,
**extras
)