Update app.py

app.py

@@ -66,15 +66,15 @@ 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)
-        ret, frame_f = capture.read(i)
+        capture.set(cv2.CAP_PROP_POS_FRAMES, i)
+        ret, frame_f = capture.read()
         #frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
         cv2.imwrite(f'{uid}-frames/{i}.png',frame_f)
         fbox.append(f'{uid}-frames/{i}.png')
         frame_num=f'Working on {i+1} of {frame_count1}'
         
         yield fbox,fbox,frame_num
-    return fbox,fbox,frame_num
+    #return fbox,fbox,frame_num
 
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size