dofaker/face_swap/__init__.py

from .inswapper import InSwapper


def get_swapper_model(name='', root=None, **kwargs):
    if name.lower() == 'inswapper':
        return InSwapper(name=name, root=root, **kwargs)
    else:
        raise UserWarning('The swapper model {} not support.'.format(name))