Update app.py

app.py

@@ -66,7 +66,9 @@ def capture_function(vid=None):
 
 def im_2_vid(images,fps):
     #width, height = Image.open(images[0]).size
-    width, height = cv2.imread(images[0]).shape()
+    this_im=cv2.imread(images[0])
+    print(this_im.shape())
+    width, height = this_im.shape()
     print (width,height)
     size = (width, height)
     codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
@@ -180,23 +182,21 @@ def other():
     
 with gr.Blocks() as app:
     with gr.Tab("Load"):
-        
-        inp_url = gr.Textbox()
-        go_btn = gr.Button("Run")
         with gr.Row():
             with gr.Column():
+                inp_url = gr.Textbox()
+                go_btn = gr.Button("Run")
                 outp_vid=gr.Video()
             with gr.Column():
                 with gr.Row():
                     frame_count=gr.Textbox(label="Frame Count",interactive = False)
                     fps=gr.Textbox(label="FPS",interactive = False)
                 outp_file=gr.Files()
-        with gr.Row():
-            start_f = gr.Textbox(label = "Start", value = "0:00:00", placeholder = "0:00:23")
-            end_f = gr.Textbox(label = "End", value = "0:00:05", placeholder = "0:00:54")
-            trim_btn=gr.Button("Trim")
-        with gr.Row():
-            out_trim=gr.Video()
+                with gr.Row():
+                    start_f = gr.Textbox(label = "Start", value = "0:00:00", placeholder = "0:00:23")
+                    end_f = gr.Textbox(label = "End", value = "0:00:05", placeholder = "0:00:54")
+                trim_btn=gr.Button("Trim")
+                out_trim=gr.Video()
 
     with gr.Tab("Frames"):
         frame_btn = gr.Button("Get Frames")
@@ -223,10 +223,10 @@ with gr.Blocks() as app:
     def echo_fn(inp):
         return inp 
     #outp_vid.change(echo_fn,outp_vid,[out_trim])
-    trim_btn.click(trim_vid,[outp_vid,start_f,end_f],out_trim)
+    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])
-    go_btn.click(dl,inp_url,[outp_vid,in_vid,outp_file,out_trim])
+    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()