try:
    from icenet import base_cnn, icenet
except Exception as ex:
    print("Failed to import keras")
    print(ex)
    base_cnn = icenet = None

MODELS = {
    'base_cnn': base_cnn,
    'icenet': icenet 
}