nishanth-saka commited on
Commit
726e57b
·
verified ·
1 Parent(s): 7b79d8a
Files changed (1) hide show
  1. app.py +3 -3
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
- # mask_img = Image.open(BytesIO(result_no_bg)).convert("RGBA")
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