Update app.py

app.py

@@ -23,45 +23,44 @@ model.overrides['max_det'] = 1000  # maximum number of detections per image
 model.to(device)
 
 
-with gr.Blocks() as app:
-    stream = gr.State()
-    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()
-       
-        return vid_cap,frame_num,frame_count,frame_fps
-        
-    def vid_play(cap,frame_num):
-        player = cv2.VideoCapture(cap)
-        assert player.isOpened() # Make sure that their is a stream. 
-        player.set(cv2.CAP_PROP_POS_FRAMES, int(frame_num))
 
-        ret, frame_bgr = player.read(int(frame_num))
-        frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)        
-        results = model.predict(frame)
-        render = render_result(model=model, image=frame, result=results[0])
-        return render
-    def fw_fn(cur,last):
-        if not int(cur)+1 > int(last):
-            next = int(cur)+1
-        else:
-            next = last
-            pass
-        return next
-    def bk_fn(cur):
-        if not int(cur)-1 < 0:
-            next = int(cur)-1
-        else:
-            next = 0
-            pass
-        return next
-     
+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()
+   
+    return vid_cap,frame_num,frame_count,frame_fps
+    
+def vid_play(cap,frame_num):
+    player = cv2.VideoCapture(cap)
+    assert player.isOpened() # Make sure that their is a stream. 
+    player.set(cv2.CAP_PROP_POS_FRAMES, int(frame_num))
+
+    ret, frame_bgr = player.read(int(frame_num))
+    frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)        
+    results = model.predict(frame)
+    render = render_result(model=model, image=frame, result=results[0])
+    return render
+def fw_fn(cur,last):
+    next = cur+1
+    if next > last:
+        next = last
+    return next
+def bk_fn(cur):
+    next = cur-1
+    if next < 0:
+        next = 0
+    return next
+
+
 
+
+
+with gr.Blocks() as app:
     with gr.Row():
         with gr.Column():
             youtube_url = gr.Textbox(label="YouTube URL",value=f"{URL}")