Update utils.py

utils.py

@@ -9,7 +9,7 @@ import torch
 import pandas as pd
 import numpy as np
 import cv2
-import yt_dlp
+# import yt_dlp
 import gradio as gr
 from ultralytics import YOLO
 
@@ -35,12 +35,12 @@ def detect_image(image_path: str, model: YOLO, conf: float, iou: float) -> np.nd
 def detect_video(video_path_or_url: str, model: YOLO, conf: float, iou: float) -> Tuple[Path, Path]:
     progress = gr.Progress()
     video_path = video_path_or_url
-    if 'youtube.com' in video_path_or_url or 'youtu.be' in video_path_or_url:
-        progress(0.001, desc='Downloading video from YouTube...')
-        ydl_opts = {'format': 'bestvideo[height<=720]'}
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            video_info_dict = ydl.extract_info(video_path_or_url, download=True)
-            video_path = ydl.prepare_filename(video_info_dict)
+    # if 'youtube.com' in video_path_or_url or 'youtu.be' in video_path_or_url:
+    #     progress(0.001, desc='Downloading video from YouTube...')
+    #     ydl_opts = {'format': 'bestvideo[height<=720]'}
+    #     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+    #         video_info_dict = ydl.extract_info(video_path_or_url, download=True)
+    #         video_path = ydl.prepare_filename(video_info_dict)
 
     cap = cv2.VideoCapture(video_path)
     num_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))