def METHOD_NAME(self, data, model):
    """A default predict_fn for PyTorch. Calls a model on data deserialized in input_fn.
    Runs prediction on GPU if cuda is available.
    Args:
        data: input data (torch.Tensor) for prediction deserialized by input_fn
        model: PyTorch model loaded in memory by model_fn
    Returns: a prediction
    """
    input_data = data.to(device)
    with torch.no_grad():
        with torch.jit.optimized_execution(True):
            output = model.forward(input_data)
    return output