Update app.py

app.py

@@ -205,22 +205,16 @@ def remove_bg(fl, count, mh, ms, md, lm):
         
         cv2.imwrite(fl, fr)
 
-        b = 15
-        element = cv2.getStructuringElement(cv2.MORPH_RECT, (2 * b + 1, 2 * b + 1), (b, b))
-        mask_e = cv2.erode(mask, element) * 255
-        mask_d = cv2.dilate(mask, element) * 127
-        mask_d[mask_e>0] = 255
-        cv2.imwrite(f"{str(count)}_trimask.png", mask_d.astype(np.uint8))
-
-        cutout(
-   # input image path
-   fl,
-   # input trimap path
-   f"{str(count)}_trimask.png",
-   # output cutout path
-   f"{str(count)}_cutout.png")
+        #b = 15
+        #element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
+        #mask_e = cv2.erode(mask, element) * 255
+        #mask_d = cv2.dilate(mask, element) * 127
+        #mask_d[mask_e>0] = 255
+        #cv2.imwrite(f"{str(count)}_trimask.png", mask_d.astype(np.uint8))
+
+        #cutout(fl, f"{str(count)}_trimask.png",  f"{str(count)}_cutout.png")
         
-        return f"{str(count)}_cutout.png"
+        return fl #f"{str(count)}_cutout.png"
     else: # even: with the flash
         fl_ = fl.split(".")[0] + "_.png"
         cv2.imwrite(fl_, fr_diff.astype(np.uint8))