Update app.py

app.py

@@ -58,7 +58,7 @@ def capture_function(vid=None,frame_count=None):
     frame_ff = cv2.cvtColor(frame_f, cv2.COLOR_BGR2RGB)
     fbox.append(frame_ff)
     capture.release()
-    return fbox
+    return fbox,fbox
 
     
 def rem_bg(inp):
@@ -117,7 +117,8 @@ with gr.Blocks() as app:
                 outp_file=gr.Files()
     with gr.Tab("Frames"):
         frame_btn = gr.Button("Get Frames")                
-        frame_gal = gr.Gallery()
+        frame_gal = gr.Gallery(columns=6)
+        frame_file = gr.Files
         
     with gr.Tab("Rem BG"):
         with gr.Row():    
@@ -132,7 +133,7 @@ with gr.Blocks() as app:
         text_output=gr.Textbox()
         url_params=gr.JSON()
     outp_vid.change(load_video,outp_vid,frame_count)
-    frame_btn.click(capture_function,[outp_vid,frame_count],frame_gal)
+    frame_btn.click(capture_function,[outp_vid,frame_count],[frame_gal,frame_files])
     rem_btn.click(rem_cv,in_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)