Update app.py

app.py

@@ -216,10 +216,13 @@ def remove_bg(fl, count, mh, ms, md, lm):
         cutout(fl, f"{str(count)}_trimask.png",  f"{str(count)}_cutout.png")
 
         a_map = cv2.imread(f"{str(count)}_cutout.png", cv2.IMREAD_UNCHANGED).astype(np.uint8)
-        a_map[:,:,3] = ((255 - a_map[:,:,3]) / 255).astype(np.uint8)
-        fr[:,:,0][a_map[:,:,3]>0] = (fr[:,:,0] * a_map[:,:,3] * 15 + 240).astype(np.uint8)
-        fr[:,:,1][a_map[:,:,3]>0] = (fr[:,:,1] * a_map[:,:,3] * 15 + 240).astype(np.uint8)
-        fr[:,:,2][a_map[:,:,3]>0] = (fr[:,:,2] * a_map[:,:,3] * 15 + 240).astype(np.uint8)
+        B, G, R, A = cv2.split(a_map)
+        alpha = A / 255
+
+        R = (255 * (1 - alpha) + R * alpha).astype(np.uint8)
+        G = (255 * (1 - alpha) + G * alpha).astype(np.uint8)
+        B = (255 * (1 - alpha) + B * alpha).astype(np.uint8)
+        fr = cv2.merge((B, G, R))
 
         cv2.imwrite(fl, fr)
         return fl #f"{str(count)}_cutout.png"