juanelot commited on
Commit
e2aec45
·
verified ·
1 Parent(s): 490d964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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, False, None, None)
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