Update app.py

app.py

@@ -174,12 +174,13 @@ def remove_bg(fl, i):
     fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
 
     if i % 2: # odd: is photo without the flash
-        fr = cv2.imread(fl_)
+        fr = cv2.imread(fl_).astype(np.uint8)
         fr = fr_diff - fr
         cv2.imwrite(fl_, fr)
     else: # even: with the flash
         fl_ = fl.split(".")[0] + "_.png"
-        cv2.imwrite(fl_, fr_diff)
+        print(fl_)
+        cv2.imwrite(fl_, fr_diff.astype(np.uint8))
         
     return fl_