def METHOD_NAME(model_name, data_shape):
    gluon_model = gluon.model_zoo.vision.METHOD_NAME(model_name, pretrained=True)
    mod, params = relay.frontend.from_mxnet(gluon_model, {"data": data_shape})
    # we want a probability so add a softmax operator
    func = mod["main"]
    func = relay.Function(
        func.params, relay.nn.softmax(func.body), None, func.type_params, func.attrs
    )
    return func, params