Spaces:
Runtime error
Runtime error
Update SegCloth.py
Browse files- SegCloth.py +1 -1
SegCloth.py
CHANGED
@@ -8,7 +8,7 @@ from transparent_background import Remover
|
|
8 |
# Initialisation du pipeline de segmentation
|
9 |
segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")
|
10 |
|
11 |
-
|
12 |
def remove_background(image):
|
13 |
remover = Remover()
|
14 |
if isinstance(image, Image.Image):
|
|
|
8 |
# Initialisation du pipeline de segmentation
|
9 |
segmenter = pipeline(model="mattmdjaga/segformer_b2_clothes")
|
10 |
|
11 |
+
#@spaces.GPU
|
12 |
def remove_background(image):
|
13 |
remover = Remover()
|
14 |
if isinstance(image, Image.Image):
|