Update app.py

app.py

@@ -58,7 +58,7 @@ def load_video(vid):
     #vid_len_up = gr.update(label = "End", value = f"{vid_len}", placeholder = "0:00:54")
     return frame_count, fps, vid_len
 
-def capture_function(vid=None):
+def capture_function(vid):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)
     frame_count = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -66,7 +66,7 @@ def capture_function(vid=None):
     cnt=0
     frame_count1= int(frame_count)
     for i in range(int(frame_count1)-1):
-        #capture.set(cv2.CAP_PROP_POS_FRAMES, i)
+        capture.set(cv2.CAP_PROP_POS_FRAMES, i)
         ret, frame_f = capture.read(i)
         #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
         cv2.imwrite(f'{uid}-frames/{i}.png',frame_f)