Update app.py

app.py

@@ -27,16 +27,16 @@ 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")
-    process = cv2.VideoCapture(vid_cap)
-    frame_num = int(process.get(cv2.CAP_PROP_POS_FRAMES))
-    frame_count = int(process.get(cv2.CAP_PROP_FRAME_COUNT))
-    frame_fps = (process.get(cv2.CAP_PROP_FPS))
-    process.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
     
 def vid_play(cap,frame_num):
-    player = cv2.VideoCapture(cap)
+    #player = cv2.VideoCapture(cap)
     assert player.isOpened() # Make sure that their is a stream. 
     player.set(cv2.CAP_PROP_POS_FRAMES, int(frame_num))
 
@@ -44,7 +44,6 @@ def vid_play(cap,frame_num):
     frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)        
     results = model.predict(frame)
     render = render_result(model=model, image=frame, result=results[0])
-    player.release()
     return render
 def fw_fn(cur,last):
     next = cur+1
@@ -57,7 +56,7 @@ def bk_fn(cur):
         next = 0
     return next
 def pl_fn(cap,cur,last,fps,pl_tog):
-    player = cv2.VideoCapture(cap)
+    #player = cv2.VideoCapture(cap)
     #assert player.isOpened() # Make sure that their is a stream. 
     player.set(cv2.CAP_PROP_POS_FRAMES, cur)
     ret, frame_bgr = player.read(int(cur))
@@ -72,6 +71,7 @@ def pl_fn(cap,cur,last,fps,pl_tog):
 
 
 with gr.Blocks() as app:
+    player = gr.State()
     with gr.Row():
         with gr.Column():
             youtube_url = gr.Textbox(label="YouTube URL",value=f"{URL}")