Update app.py

app.py

@@ -8,7 +8,7 @@ from ultralyticsplus import YOLO, render_result
 
 model = YOLO('ultralyticsplus/yolov8s')
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-URL = "https://www.youtube.com/watch?v=dQw4w9WgXcQ" #URL to parse
+URL = "https://www.youtube.com/watch?v=6NBwbKMyzEE" #URL to parse
 
 # set model parameters
 model.overrides['conf'] = 0.25  # NMS confidence threshold
@@ -19,15 +19,17 @@ model.to(device)
 
 
 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")
     global player
+    player.release()
     player = cv2.VideoCapture(vid_cap)
     frame_num = int(player.get(cv2.CAP_PROP_POS_FRAMES))
     frame_count = int(player.get(cv2.CAP_PROP_FRAME_COUNT))
     frame_fps = (player.get(cv2.CAP_PROP_FPS))
-    #process.release()
-    return vid_cap,frame_num,frame_count,frame_fps
+    tog = 0
+    return vid_cap,frame_num,frame_count,frame_fps,tog
     
 def vid_play(cap,frame_num):
     assert player.isOpened() # Make sure that their is a stream. 
@@ -93,7 +95,7 @@ with gr.Blocks() as app:
     cur_frame.change(pl_fn,[output_win,cur_frame,total_frames,fps_frames,pl_tog],[det_win,cur_frame],show_progress=False)
     bk.click(bk_fn,[cur_frame],cur_frame,show_progress=False)
     fw.click(fw_fn,[cur_frame,total_frames],cur_frame,show_progress=False)
-    load_button.click(load,youtube_url,[output_win,cur_frame,total_frames,fps_frames])
+    load_button.click(load,youtube_url,[output_win,cur_frame,total_frames,fps_frames,pl_tog])
     #run_button.click(vid_play, [output_win,cur_frame], det_win) 
 
 app.queue(concurrency_count=10).launch()