models/apis/__init__.py

from .train import train_model

__all__ = [
    'train_model'
]