Update app.py

app.py

@@ -44,6 +44,7 @@ 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
@@ -55,7 +56,17 @@ def bk_fn(cur):
     if next < 0:
         next = 0
     return next
-
+def pl_fn(cap,cur,last,fps,pl_tog):
+        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))
+        frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
+    if pl_tog ==1:
+        cur+=1
+    else:
+        cur = cur        
+    return frame,cur
 
 
 
@@ -77,9 +88,12 @@ with gr.Blocks() as app:
                 pl = gr.Button("Play")
                 fw = gr.Button(">")
             det_win = gr.Image(source="webcam", streaming=True)
+    with gr.Row():
+        pl_tog=gr.Number()
+    pl.click(pl_fn,[output_win,cur_frame,total_frames,fps_frames,pl_tog],det_win,every=0.1)
     bk.click(bk_fn,[cur_frame],cur_frame)
     fw.click(fw_fn,[cur_frame,total_frames],cur_frame)
     load_button.click(load,youtube_url,[output_win,cur_frame,total_frames,fps_frames])
     run_button.click(vid_play, [output_win,cur_frame], det_win) 
 
-app.launch(enable_queue=False)
+app.queue(concurrency_count=10).launch()