Update app.py

app.py

@@ -24,12 +24,14 @@ import sys
 from enum import Enum
 from typing import Any, Dict, List, Optional, Tuple
 
+import os
 import cv2
 import gradio as gr
 import innertube
 import numpy as np
 from PIL import Image
 from ultralytics import YOLO
+
 import yt_dlp
 
 logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
@@ -90,7 +92,7 @@ class SearchService:
 
     @staticmethod
     def get_stream(youtube_url: str) -> Optional[str]:
-        """Retrieves the livestream URL for a given YouTube video URL using yt-dlp.
+        """Retrieves the livestream URL for a given YouTube video URL using yt-dlp with cookies.
 
         :param youtube_url: The URL of the YouTube video.
         :type youtube_url: str
@@ -103,11 +105,14 @@ class SearchService:
             'no_warnings': True,
             'force_generic_extractor': False,
             'skip_download': True,
+            # Include the cookies file
+            'cookiefile': 'cookies.txt',
         }
         try:
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info_dict = ydl.extract_info(youtube_url, download=False)
                 if info_dict.get('is_live'):
+                    # Get the live stream URL
                     live_url = info_dict.get('url')
                     if live_url:
                         logging.debug(f"Found livestream URL: {live_url}")
@@ -127,7 +132,7 @@ INITIAL_STREAMS = SearchService.search("world live cams", SearchFilter.LIVE)
 class LiveYouTubeObjectDetector:
     def __init__(self):
         logging.getLogger().setLevel(logging.DEBUG)
-        self.model = YOLO("yolo11n.pt")
+        self.model = YOLO("yolov8n.pt")
         self.streams = INITIAL_STREAMS
 
         # Gradio UI
@@ -241,4 +246,4 @@ class LiveYouTubeObjectDetector:
         app.queue().launch(show_api=False, debug=True)
 
 if __name__ == "__main__":
-    LiveYouTubeObjectDetector().render()
+    LiveYouTubeObjectDetector().render()