Update app.py

app.py

@@ -6,6 +6,13 @@ from pytube import YouTube
 from ultralyticsplus import YOLO, render_result
 
 #from torch import hub # Hub contains other models like FasterRCNN
+model = YOLO('ultralyticsplus/yolov8s')
+
+# set model parameters
+model.overrides['conf'] = 0.25  # NMS confidence threshold
+model.overrides['iou'] = 0.45  # NMS IoU threshold
+model.overrides['agnostic_nms'] = False  # NMS class-agnostic
+model.overrides['max_det'] = 1000  # maximum number of detections per image
 
 URL = "https://www.youtube.com/watch?v=dQw4w9WgXcQ" #URL to parse
 #play = pafy.new(_URL).streams[-1] #'-1' means read the lowest quality of video.
@@ -19,14 +26,7 @@ class Capvid:
  
     
     # load model
-    model = YOLO('ultralyticsplus/yolov8s')
-    
-    # set model parameters
-    model.overrides['conf'] = 0.25  # NMS confidence threshold
-    model.overrides['iou'] = 0.45  # NMS IoU threshold
-    model.overrides['agnostic_nms'] = False  # NMS class-agnostic
-    model.overrides['max_det'] = 1000  # maximum number of detections per image
-    
+
     # set image
     #image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
     
@@ -102,6 +102,7 @@ class Capvid:
             print(f"Frames Per Second : {fps}")
             out.write(frame) # Write the frame onto the output.
             ret, frame = player.read() # Read next frame.
+            return frame
 
 def vid_play():
     Capvid()