Update app.py

app.py

@@ -101,25 +101,6 @@ def detect_video(video):
     # Release resources
     cap.release()
 
-
-def show_preds_video(video_path):
-    cap = cv2.VideoCapture(video_path)
-    while(cap.isOpened()):
-        ret, frame = cap.read()
-        if ret:
-            frame_copy = frame.copy()
-            outputs = detection_model.predict(source=frame)
-            results = outputs[0].cpu().numpy()
-            for i, det in enumerate(results.boxes.xyxy):
-                cv2.rectangle(
-                    frame_copy,
-                    (int(det[0]), int(det[1])),
-                    (int(det[2]), int(det[3])),
-                    color=(0, 0, 255),
-                    thickness=2,
-                    lineType=cv2.LINE_AA
-                )
-            yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)
  
 inputs_video = [
     gr.components.Video(type="filepath", label="Input Video"),
@@ -177,7 +158,7 @@ c = os.path.join(os.path.dirname(__file__), "data/c.mp4")  # Video
 # )
 
 stt_demo = gr.Interface(
-    fn=show_preds_video,
+    fn=detect_video,
     inputs=inputs_video,
     outputs=outputs_video,
     examples=[