Update app.py

app.py

@@ -4,6 +4,11 @@ import cv2 # opencv2 package for python.
 import torch
 from pytube import YouTube
 from ultralyticsplus import YOLO, render_result
+#from imageai.Detection import ObjectDetection
+
+
+obj_detect.setModelPath(r"C:/Datasets/yolo.h5")
+obj_detect.loadModel()
 
 #from torch import hub # Hub contains other models like FasterRCNN
 model = YOLO('ultralyticsplus/yolov8s')
@@ -109,15 +114,40 @@ def vid_play(vid_cap):
         ret, frame = player.read() # Read next frame.
         return out
 
-
- 
-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")
-
-    return vid_cap
-
 with gr.Blocks() as app:
+    def vid_play2(vid_cap):
+        stream = cv2.VideoCapture(vid_cap)
+        
+        player = stream #Get your video stream.
+        assert player.isOpened() # Make sure that their is a stream. 
+        #Below code creates a new video writer object to write our
+        #output stream.
+        out_vid = ("vid_tmp.avi")
+        x_shape = int(player.get(cv2.CAP_PROP_FRAME_WIDTH))
+        y_shape = int(player.get(cv2.CAP_PROP_FRAME_HEIGHT))
+        four_cc = cv2.VideoWriter_fourcc(*"MJPG") #Using MJPEG codex
+        out = cv2.VideoWriter(out_vid, four_cc, 20, \
+                              (x_shape, y_shape)) 
+        ret, frame = player.read() # Read the first frame.
+        
+        while True: # Run until stream is out of frames
+            start_time = time.time() # We would like to measure the FPS.
+            results = score_frame(frame) # Score the Frame
+            frame = plot_boxes(results, frame) # Plot the boxes.
+            end_time = time.time()
+            fps = 1/np.round(end_time - start_time, 3) #Measure the FPS.
+            print(f"Frames Per Second : {fps}")
+            out.write(frame) # Write the frame onto the output.
+            ret, frame = player.read() # Read next frame.
+            return gr.Video(out)
+   
+     
+    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")
+    
+        return vid_cap
+
     youtube_url = gr.Textbox(label="YouTube URL",value=f"{URL}")
     load_button = gr.Button("Load Video")
     run_button = gr.Button()
@@ -125,6 +155,6 @@ with gr.Blocks() as app:
         output_win = gr.Video()
         det_win = gr.Video()
     load_button.click(load,youtube_url,output_win)
-    run_button.click(vid_play, [output_win], det_win) 
+    run_button.click(vid_play2, [output_win], det_win) 
 
 app.launch(enable_queue=False)