custom_classifier/__init__.py

from .configuration import CustomModelConfig
from .model import CustomClassifier