Spaces:
Runtime error
Runtime error
Commit
Β·
432c58e
1
Parent(s):
fda1d3e
update: minor changes
Browse files
app.py
CHANGED
@@ -23,15 +23,6 @@ def clear_func():
|
|
23 |
gc.collect()
|
24 |
|
25 |
|
26 |
-
def process_mask(mask):
|
27 |
-
mask = mask.convert("L")
|
28 |
-
mask = np.array(mask)
|
29 |
-
|
30 |
-
mask = np.where(mask > 128, 255, 0).astype(np.uint8)
|
31 |
-
|
32 |
-
return Image.fromarray(mask)
|
33 |
-
|
34 |
-
|
35 |
@spaces.GPU
|
36 |
def clothing_try_on(image, mask):
|
37 |
jewellery_mask = Image.fromarray(
|
@@ -85,7 +76,7 @@ def clothing_try_on(image, mask):
|
|
85 |
|
86 |
def base64_to_image(base64_str):
|
87 |
image_data = base64.b64decode(base64_str)
|
88 |
-
image = Image.open(BytesIO(image_data))
|
89 |
return image
|
90 |
|
91 |
|
|
|
23 |
gc.collect()
|
24 |
|
25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
@spaces.GPU
|
27 |
def clothing_try_on(image, mask):
|
28 |
jewellery_mask = Image.fromarray(
|
|
|
76 |
|
77 |
def base64_to_image(base64_str):
|
78 |
image_data = base64.b64decode(base64_str)
|
79 |
+
image = Image.open(BytesIO(image_data)).convert("RGB")
|
80 |
return image
|
81 |
|
82 |
|