models/__init__.py

from utils.registry import MODEL_REGISTRY

# Register additional modules
import additional_modules.gfpgan.gfpganv1_clean_arch



def get_model(opt):

    MODEL_REGISTRY.scan_and_register()
    return MODEL_REGISTRY.get(opt['model_class'])(**opt['params'])