Update app.py

app.py

@@ -65,8 +65,8 @@ def capture_function(vid=None):
     fbox=[]
     cnt=0
     frame_count1= int(frame_count)
-    for i in range(int(frame_count1)-2):
-        capture.set(cv2.CAP_PROP_POS_FRAMES, i)
+    for i in range(int(frame_count1)-1):
+        #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)
@@ -74,7 +74,7 @@ def capture_function(vid=None):
         frame_num=f'Working on {i+1} of {frame_count1}'
         
         yield fbox,fbox,frame_num
-    yield fbox,fbox,frame_num
+    return fbox,fbox,frame_num
 
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size