Spaces:
Sleeping
Sleeping
REVERT
Browse files
app.py
CHANGED
|
@@ -118,8 +118,7 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
|
|
| 118 |
|
| 119 |
# Get RGBA from bgrem
|
| 120 |
result_no_bg = bgrem_remove(base_bytes)
|
| 121 |
-
|
| 122 |
-
|
| 123 |
|
| 124 |
# Extract alpha and clean edges
|
| 125 |
mask_alpha = np.array(mask_img)[:, :, 3].astype(np.float32) / 255.0
|
|
@@ -144,7 +143,8 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
|
|
| 144 |
alpha_channel = (mask_blurred * 255).astype(np.uint8)
|
| 145 |
pattern_rgba = np.dstack((pattern_rgb, alpha_channel))
|
| 146 |
|
| 147 |
-
return Image.fromarray(pattern_rgba, mode="RGBA")
|
|
|
|
| 148 |
|
| 149 |
# ===============================
|
| 150 |
# WRAPPER: ACCEPT BYTES OR BASE64
|
|
|
|
| 118 |
|
| 119 |
# Get RGBA from bgrem
|
| 120 |
result_no_bg = bgrem_remove(base_bytes)
|
| 121 |
+
mask_img = Image.open(BytesIO(result_no_bg)).convert("RGBA")
|
|
|
|
| 122 |
|
| 123 |
# Extract alpha and clean edges
|
| 124 |
mask_alpha = np.array(mask_img)[:, :, 3].astype(np.float32) / 255.0
|
|
|
|
| 143 |
alpha_channel = (mask_blurred * 255).astype(np.uint8)
|
| 144 |
pattern_rgba = np.dstack((pattern_rgb, alpha_channel))
|
| 145 |
|
| 146 |
+
# return Image.fromarray(pattern_rgba, mode="RGBA")
|
| 147 |
+
return Image.fromarray(pattern_rgba)
|
| 148 |
|
| 149 |
# ===============================
|
| 150 |
# WRAPPER: ACCEPT BYTES OR BASE64
|