Spaces:
Running
Running
si.qian commited on
Commit ·
51c357d
1
Parent(s): af177c5
solve bug
Browse files
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 |
-
|
| 36 |
-
session = new_session(
|
|
|
|
| 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 |
|