Update app.py

app.py

@@ -173,14 +173,14 @@ def remove_bg(fl, i):
     
     fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
 
-    fl_ = fl.split(".")[0] + "_.png"
-    if float(int(i/2)) == i/2: # is photo with the flash
-        print(i)
-        cv2.imwrite(fl_, fr_diff)
-    else: # is without
+    if i % 2: # odd: is photo without the flash
         fr = cv2.imread(fl_)
         fr = fr_diff - fr
         cv2.imwrite(fl_, fr)
+    else: # even: with the flash
+        fl_ = fl.split(".")[0] + "_.png"
+        cv2.imwrite(fl_, fr_diff)
+        
     return fl_
 
 def sharpest(fl, i):