def METHOD_NAME(model, model_type="Sequential"):
    # Drop first layer (Input layer) of Functional model
    if model_type == "Functional":
        return model.layers[1:]
    elif model_type == "Sequential":
        return model.layers