Update app.py

app.py

@@ -216,8 +216,10 @@ 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.float16)
-        fr[a_map[:,:,3]>0] = (fr * a_map[:,:,3] * 15 + 240).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)
 
         cv2.imwrite(fl, fr)
         return fl #f"{str(count)}_cutout.png"