def predict_forward_fn(model, batch):
    """Use model and batch data to predict ner tag."""
    word = batch["text_tag"]["data"]
    char = batch["char_tag"]["data"]
    word_masks = batch["text_tag"]["masks"][0]
    output = model.decode(input_word=word, input_char=char, mask=word_masks)
    output = output.numpy()
    return {"output_tag": output}