Update app.py

app.py

@@ -118,10 +118,12 @@ with gr.Blocks() as app:
     def load(URL):
         yt = YouTube(URL)
         vid_cap = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().last().download(filename="tmp.mp4")
-        #stream = cv2.VideoCapture(vid_cap)
-        return vid_cap
+        stream = cv2.VideoCapture(vid_cap)
+        frame_num = stream.get(cv2.CAP_PROP_POS_FRAMES)
+        print (frame_num)
+        return stream,frame_num
         
-    def vid_play2(cap):
+    def vid_play2(cap,frame_num):
         player = cv2.VideoCapture(cap)
         
         #player = stream #Get your video stream.
@@ -133,8 +135,8 @@ with gr.Blocks() as app:
         #y_shape = int(player.get(cv2.CAP_PROP_FRAME_HEIGHT))
         #four_cc = cv2.VideoWriter_fourcc(*"MJPG") #Using MJPEG codex
         #out = cv2.VideoWriter(out_vid, four_cc, 20,(x_shape, y_shape)) 
-        frame_num = player.get(cv2.CAP_PROP_POS_FRAMES)
-        print (frame_num)
+        player.set(cv2.CAP_PROP_POS_FRAMES, frame_num)
+
         ret, frame = player.read(frame_num)
         results = model.predict(frame)
 
@@ -155,11 +157,13 @@ with gr.Blocks() as app:
 
     youtube_url = gr.Textbox(label="YouTube URL",value=f"{URL}")
     load_button = gr.Button("Load Video")
-    run_button = gr.Button()
+    with gr.Row():
+        run_button = gr.Button()
+        cur_frame = gr.Number(value=0)
     with gr.Row():
         output_win = gr.Video()
         det_win = gr.Image(source="webcam", streaming=True)
-    load_button.click(load,youtube_url,output_win)
-    run_button.click(vid_play2, [output_win], det_win) 
+    load_button.click(load,youtube_url,[output_win,cur_frame])
+    run_button.click(vid_play2, [output_win,cur_frame], det_win) 
 
 app.launch(enable_queue=False)