si.qian commited on
Commit
51c357d
·
1 Parent(s): af177c5

solve bug

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -6,7 +6,7 @@ import os, shutil
6
  import gradio as gr
7
  from PIL import Image
8
  import sys, os
9
- from rembg import new_session,remove
10
  from utils import functions
11
  from utils.functions import complete, clean, clean_by_name, get_random_name
12
 
@@ -31,9 +31,10 @@ def run_rembg(img):
31
 
32
  return cropped_image
33
 
34
- def run_rembg_withmodel(img, model):
35
- model_name = model
36
- session = new_session(model_name)
 
37
  output = remove(img, session=session)
38
  output_pil = Image.fromarray(output)
39
 
 
6
  import gradio as gr
7
  from PIL import Image
8
  import sys, os
9
+ from rembg import new_session, remove
10
  from utils import functions
11
  from utils.functions import complete, clean, clean_by_name, get_random_name
12
 
 
31
 
32
  return cropped_image
33
 
34
+ def run_rembg_withmodel(img, model):
35
+ logger.debug(f"model name : {model}")
36
+ session = new_session("isnet-general-use")
37
+ logger.debug(f"session name : {session}")
38
  output = remove(img, session=session)
39
  output_pil = Image.fromarray(output)
40