Update app.py

app.py

@@ -50,13 +50,14 @@ 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')
+        cv2.imwrite(f'{uid}-{i}.png',frame_ff)
         fbox.append(f'{uid}-{i}.png')
         
     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)
-    fbox.append(frame_ff)
+    cv2.imwrite(f'{uid}-{int(frame_count)-1}.png',frame_ff)
+    fbox.append(f'{uid}-{int(frame_count)-1}.png')
     capture.release()
     return fbox,fbox