Update app.py

app.py

@@ -137,8 +137,8 @@ def infer(f_in, interpolation, fps_output):
     return final_vid, files
 
 
-def remove_bg(fr):
-    frame = cv2.imread(fr).astype(np.uint8)
+def remove_bg(fl):
+    frame = cv2.imread(fl).astype(np.uint8)
     
     b = 5
     #subtract background (get scene with shadow) and correct hue against light
@@ -180,7 +180,9 @@ def remove_bg(fr):
     frame_[mask>0] = (125,125,125)
     frame_[frame_[:,:,0]>=142] = (255,255,255)
 
-    return frame#, frame_
+    cv2.imwrite(fl, frame) #frame_
+
+    return fl
 
 def logscale(linear):
     return int(math.pow(2, linear))