Update app.py

app.py

@@ -52,7 +52,7 @@ try:
     import numpy as np
     import streamlink
     from PIL import Image
-    from super_gradients.training import models
+    from ultralytics import YOLO
 except ImportError:
     install_requirements()
     import cv2
@@ -61,8 +61,7 @@ except ImportError:
     import numpy as np
     import streamlink
     from PIL import Image
-    from super_gradients.training import models
-
+    from ultralytics import YOLO
 
 logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
 
@@ -252,7 +251,7 @@ class LiveYouTubeObjectDetector:
     def __init__(self):
         """Initializes the LiveYouTubeObjectDetector with YOLO model and UI components."""
         logging.getLogger().setLevel(logging.DEBUG)
-        self.model = models.get("yolo_nas_l", pretrained_weights="coco")
+        self.model = YOLO("yolov8x.pt")
         self.streams = INITIAL_STREAMS
 
         # Gradio UI