Update app.py

app.py

@@ -71,9 +71,10 @@ def capture_function(vid=None):
         #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-1}'
+        frame_num=f'Working on {i+1} of {frame_count1}'
         
         yield fbox,fbox,frame_num
+    yield fbox,fbox,frame_num
 
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size
@@ -109,7 +110,7 @@ def rem_bg(vid):
         fbox2.append(f'{uid}-rembg/{i}.png')
         frame_num=f'Working on {i+1} of {frame_count1-1}'
         yield fbox2,frame_num,None
-    frame_num=f'Done: {i+1} of {frame_count1-1}'
+    frame_num=f'Done: {i+1} of {frame_count1}'
     out_vid,_file = im_2_vid(fbox2, fps)
     yield (fbox2,frame_num,out_vid)