Update app.py

app.py

@@ -1,77 +1,34 @@
 """
-Реальное время: распознавание объектов с веб-камеры через Gradio и YOLOv8.
-Работает с Gradio 3.49.3
+Реальное время: распознавание объектов с веб-камеры через Gradio 5.x и YOLOv8.
+Работает на Hugging Face Spaces без проблем с версией Gradio.
 """
 
 import numpy as np
 from PIL import Image
 import gradio as gr
+from ultralytics import YOLO
 
-try:
-    from ultralytics import YOLO
-except Exception as e:
-    YOLO = None
-    print("Ultralytics import failed:", e)
+# Загружаем YOLO модель (лучше легкую для портфолио)
+model = YOLO("yolov8n.pt")
 
-# Кэширование модели
-_MODEL = {"obj": None}
-
-def load_model(model_name: str = "yolov8n.pt"):
-    """Загружает YOLO-модель"""
-    if YOLO is None:
-        raise RuntimeError("Пакет ultralytics не найден. Проверь requirements.txt.")
-    if _MODEL["obj"] is None:
-        _MODEL["obj"] = YOLO(model_name)
-    return _MODEL["obj"]
-
-def annotate_frame(frame: np.ndarray, conf: float = 0.25):
-    """Выполняет детекцию и возвращает кадр с аннотациями"""
+def detect(frame: np.ndarray):
+    """
+    Получает кадр с веб-камеры, выполняет детекцию YOLOv8 и возвращает аннотированный кадр.
+    """
     if frame is None:
         return None
-    model = load_model()
-    results = model.predict(source=frame, imgsz=640, conf=conf, verbose=False)
+    results = model.predict(source=frame, imgsz=640, conf=0.25, verbose=False)
     annotated = results[0].plot()
-    return np.array(annotated)
-
-def detect_live(image, conf):
-    """Обработчик Gradio для стриминга с веб-камеры"""
-    if image is None:
-        return None
-    annotated = annotate_frame(image, conf)
-    return Image.fromarray(annotated)
-
-def main():
-    title = "🎥 Реальное время: распознавание объектов (YOLOv8)"
-    description = "Потоковое детектирование объектов с веб-камеры. Используется модель `yolov8n.pt`."
-
-    with gr.Blocks() as demo:
-        gr.Markdown(f"# {title}\n\n{description}")
-
-        with gr.Row():
-            with gr.Column(scale=2):
-                # Для Gradio 3.49.3 — рабочий способ захвата веб-камеры
-                webcam = gr.Image(source="webcam", streaming=True, label="Веб-камера")
-                conf = gr.Slider(0.0, 1.0, value=0.25, step=0.01, label="Порог достоверности")
-                model_path = gr.Textbox(value="yolov8n.pt", label="Модель", visible=False)
-
-            with gr.Column(scale=1):
-                output = gr.Image(label="Результат детекции", interactive=False)
-                start_btn = gr.Button("Загрузить модель")
-                status = gr.Textbox(value="Модель не загружена", label="Статус", interactive=False)
-
-        # Кнопка загрузки модели
-        def load_event(spec):
-            try:
-                load_model(spec)
-                return f"Модель загружена: {spec}"
-            except Exception as e:
-                return f"Ошибка: {e}"
+    return annotated
 
-        start_btn.click(load_event, inputs=[model_path], outputs=[status])
-        webcam.change(detect_live, inputs=[webcam, conf], outputs=[output])
+with gr.Blocks() as demo:
+    gr.Markdown("### 🎥 YOLOv8: Распознавание объектов с веб-камеры в реальном времени")
 
-        demo.launch()
+    with gr.Row():
+        webrtc = gr.Video(label="Веб-камера", source="webcam", streaming=True)
+        output = gr.Image(label="Результат детекции")
 
-if __name__ == "__main__":
-    main()
+    # Подключаем обработчик
+    webrtc.change(detect, inputs=[webrtc], outputs=[output])
 
+demo.launch()