Update app.py

app.py

@@ -50,9 +50,9 @@ def capture_function(vid=None,frame_count=None):
         capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
         frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
+        im_file = cv2.imwrite(f'{uid}-{i}.png')
+        fbox.append(f'{uid}-{i}.png')
         
-        fbox.append(frame_ff)
-
     capture.set(cv2.CAP_PROP_POS_FRAMES, int(frame_count)-1)
     ret, frame_f = capture.read()
     frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
@@ -66,6 +66,7 @@ def rem_bg(inp):
     ret, frame = cap.read()
     
     out = rem_bg()
+    out = cv2.VideoWriter('video.avi',cv2.VideoWriter_fourcc(*'DIVX'), 15, size)