Update app.py

app.py

@@ -151,8 +151,8 @@ with gr.Blocks() as app:
         text_output=gr.Textbox()
         url_params=gr.JSON()
     outp_vid.change(load_video,outp_vid,[frame_count,fps])
-    frame_btn.click(capture_function,[outp_vid,frame_count],[frame_gal,frame_file])
-    rem_btn.click(rem_bg,[outp_vid,frame_count],rem_vid)
+    frame_btn.click(capture_function,[outp_vid],[frame_gal,frame_file])
+    rem_btn.click(rem_bg,[outp_vid],rem_vid)
     go_btn.click(dl,inp_url,[outp_vid,in_vid,outp_file])
     app.load(fn=predict, inputs=[text_input,url_params], outputs=[text_output,text_input],_js=load_js)
 app.launch()