src/models/__init__.py

from .model import Summarization
from .train_model import train_model
from .predict_model import predict_model
from .evaluate_model import evaluate_model