juanelot commited on
Commit
7388e1c
·
verified ·
1 Parent(s): e2aec45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,10 +18,11 @@ def predict(image):
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
25
 
26
  with gr.Blocks(title="Remove background") as app:
27
  gr.HTML("<center><h1>Background Remover</h1></center>")
 
18
  matting = (0, 0, 0) # Valores predeterminados para matting
19
  bg_color = (0, 0, 0) # Color de fondo predeterminado (negro)
20
  try:
21
+ result = remove(session, image, smoot, matting, bg_color)
22
+ return result
23
  except ValueError as err:
24
  logging.error(err)
25
+ return None
26
 
27
  with gr.Blocks(title="Remove background") as app:
28
  gr.HTML("<center><h1>Background Remover</h1></center>")