Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,11 @@ default_model = "U2NET"
|
|
14 |
|
15 |
def predict(image):
|
16 |
session = new_session(remove_bg_models[default_model])
|
|
|
|
|
|
|
17 |
try:
|
18 |
-
return remove(session, image,
|
19 |
except ValueError as err:
|
20 |
logging.error(err)
|
21 |
return make_label(str(err)), None
|
|
|
14 |
|
15 |
def predict(image):
|
16 |
session = new_session(remove_bg_models[default_model])
|
17 |
+
smoot = False
|
18 |
+
matting = (0, 0, 0) # Valores predeterminados para matting
|
19 |
+
bg_color = (0, 0, 0) # Color de fondo predeterminado (negro)
|
20 |
try:
|
21 |
+
return remove(session, image, smoot, matting, bg_color)
|
22 |
except ValueError as err:
|
23 |
logging.error(err)
|
24 |
return make_label(str(err)), None
|