Update app.py

app.py

@@ -66,8 +66,8 @@ 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()
+        #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)
         fbox.append(f'{uid}-frames/{i}.png')
@@ -75,6 +75,7 @@ def capture_function(vid=None):
         
         yield fbox,fbox,frame_num
     #return fbox,fbox,frame_num
+    yield (fbox,fbox,frame_num)
 
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size