doctr/models/kie_predictor/__init__.py

from doctr.file_utils import is_tf_available

if is_tf_available():
    from .tensorflow import *
else:
    from .pytorch import *  # type: ignore[assignment]