def find_framework(model_module):
    """
    Find the deep learning framework used from a module - based on items on the module

    Parameters
    -----------
    model_module - module
        Module that contains the core model logic

    Returns
    ------------
    str - with the framework name
    """

    tensorflow_imports = len([module for module in dir(model_module) if 'tensorflow' in module or 'tf' in module])
    torch_imports = len([module for module in dir(model_module) if 'torch' in module])

    if tensorflow_imports or torch_imports:
        if tensorflow_imports > torch_imports:
            return 'tensorflow'
        elif torch_imports > tensorflow_imports:
            return 'pytorch'
        else:
            return 'tensorflow'
    else:
        return 'none'