Spaces:
Running
Running
si.qian
commited on
Commit
·
3bfcd29
1
Parent(s):
db9435e
solve bug
Browse files
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(
|
| 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)
|