Spaces:
Runtime error
Runtime error
Update SegCloth.py
Browse files- SegCloth.py +1 -1
SegCloth.py
CHANGED
@@ -20,7 +20,7 @@ def encode_image_to_base64(image):
|
|
20 |
image.save(buffered, format="PNG")
|
21 |
return base64.b64encode(buffered.getvalue()).decode('utf-8')
|
22 |
|
23 |
-
def
|
24 |
# Segment image
|
25 |
segments = segmenter(img)
|
26 |
|
|
|
20 |
image.save(buffered, format="PNG")
|
21 |
return base64.b64encode(buffered.getvalue()).decode('utf-8')
|
22 |
|
23 |
+
def segment_clothing(img, clothes=["Hat", "Upper-clothes", "Skirt", "Pants", "Dress", "Belt", "Left-shoe", "Right-shoe", "Scarf"]):
|
24 |
# Segment image
|
25 |
segments = segmenter(img)
|
26 |
|