app/models/__init__.py

"""Модели данных и ML модель"""

from app.models.wagon_model import WagonClassifier, get_classifier
from app.models.schemas import PredictionResponse, ErrorResponse, HealthResponse

__all__ = [
    'WagonClassifier',
    'get_classifier',
    'PredictionResponse',
    'ErrorResponse',
    'HealthResponse'
]