# Pydantic models and YOLOv5 wrapper