freealise commited on
Commit
1799cdd
·
verified ·
1 Parent(s): 15db140

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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_)