Build the Future withAI Vision: Глаза вашего бизнеса в MAX

import torch
from ultralytics import YOLO

class VisualExpert:
    def __init__(self):
        self.model = YOLO('custom_weights_2026.pt')
        self.ocr = load_ocr_engine('ru')

    def process_frame(self, image_stream):
        # 1. Детекция объектов
        results = self.model.predict(image_stream, conf=0.7)
        labels = results[0].boxes.cls.tolist()

        # 2. Если найден документ — запускаем OCR
        if 0 in labels: # ID для 'Document'
            data = self.ocr.recognize(image_stream)
            return {"type": "doc", "payload": data}
        
        # 3. Анализ сцены и поиск соответствий в базе
        return {"type": "scene", "detections": labels}