Update app.py

app.py

@@ -86,7 +86,7 @@ def rem_bg(vid):
         print (i)
         cv2.imwrite(f'{uid}-rembg/{i}.png',out)
         fbox2.append(f'{uid}-rembg/{i}.png')
-        frame_num=f'Working on {i+1} of {frame_count}'
+        frame_num=f'Working on {i+1} of {frame_count1-1}'
         yield fbox2,frame_num,None
 
     out_vid = im_2_vid(fbox2, fps)
@@ -196,6 +196,7 @@ with gr.Blocks() as app:
                 #rem_vid=gr.Video()
                 frame_num=gr.Textbox(label="Progress")
                 rem_vid=gr.Gallery(columns=6)
+        rem_bg_vid=gr.Video()        
     with gr.Row(visible=False):
         text_input=gr.Textbox()
         text_output=gr.Textbox()
@@ -207,7 +208,7 @@ with gr.Blocks() as app:
     trim_btn.click(trim_vid,[outp_vid,start_f,end_f],[out_trim,in_vid])
     outp_vid.change(load_video,outp_vid,[frame_count,fps]).then(echo_fn,outp_vid,[out_trim])
     frame_btn.click(capture_function,[out_trim],[frame_gal,frame_file])
-    rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num])
+    rem_btn.click(rem_bg,[out_trim],[rem_vid,frame_num,rem_bg_vid])
     go_btn.click(dl,inp_url,[outp_vid,outp_file,out_trim])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()