Update app.py
Browse files
app.py
CHANGED
@@ -173,8 +173,8 @@ def remove_bg(fl, i):
|
|
173 |
|
174 |
fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
|
175 |
|
|
|
176 |
if int(i/2) == i/2: # is photo with the flash
|
177 |
-
fl_ = fl.split(".")[0] + "_.png"
|
178 |
cv2.imwrite(fl_, fr_diff)
|
179 |
else: # is without
|
180 |
fr = cv2.imread(fl_)
|
|
|
173 |
|
174 |
fr_diff = cv2.dilate(cv2.erode(fr_diff, element), element)
|
175 |
|
176 |
+
fl_ = fl.split(".")[0] + "_.png"
|
177 |
if int(i/2) == i/2: # is photo with the flash
|
|
|
178 |
cv2.imwrite(fl_, fr_diff)
|
179 |
else: # is without
|
180 |
fr = cv2.imread(fl_)
|