si.qian commited on
Commit
3bfcd29
·
1 Parent(s): db9435e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def run_rembg(img):
34
  def run_rembg_withmodel(img, model):
35
  logger.debug(f"model name : {model}")
36
  logger.debug(f"model name : {type(model)}")
37
- session = new_session("isnet-general-use")
38
  logger.debug(f"session name : {session}")
39
  output = remove(img, session=session)
40
  output_pil = Image.fromarray(output)
 
34
  def run_rembg_withmodel(img, model):
35
  logger.debug(f"model name : {model}")
36
  logger.debug(f"model name : {type(model)}")
37
+ session = new_session(model)
38
  logger.debug(f"session name : {session}")
39
  output = remove(img, session=session)
40
  output_pil = Image.fromarray(output)