Update app.py

app.py

@@ -1,11 +1,11 @@
-import gradio as gr
-from gradio_webrtc import VideoTransformerBase, webrtc_streamer
-from ultralytics import YOLO
 import cv2
+from ultralytics import YOLO
+import fastrtc
+import gradio as gr
 
 model = YOLO("yolov8n.pt")
 
-class YOLOTransformer(VideoTransformerBase):
+class YOLOTransformer(fastrtc.VideoTransformer):
     def transform(self, frame):
         results = model.predict(source=frame, imgsz=640, conf=0.25, verbose=False)
         annotated = results[0].plot()
@@ -13,7 +13,7 @@ class YOLOTransformer(VideoTransformerBase):
 
 with gr.Blocks() as demo:
     gr.Markdown("### 🎥 YOLOv8: Распознавание объектов с веб-камеры")
-    webrtc_streamer(key="yolo", video_transformer_factory=YOLOTransformer)
+    fastrtc.webrtc_streamer(key="yolo", video_transformer_factory=YOLOTransformer)
 
 demo.launch()