get_entity_spans and model.sample output in entity linking about genre HOT 4 CLOSED

def _get_entity_spans(
    model,
    input_sentences,
    prefix_allowed_tokens_fn,
    redirections=None,
):
    output_sentences = model.sample(
        input_sentences,
        prefix_allowed_tokens_fn=prefix_allowed_tokens_fn,
    )

    output_sentences = [e[0]["text"] for e in output_sentences]
   
    return get_entity_spans_finalize(
        input_sentences, output_sentences, redirections=redirections
    )

sentences = ["Tired of the lies? Tired of the spin? Are you ready to hear the hard-hitting truth in comprehensive, conservative, principled fashion? The Ben Shapiro Show brings you all the news you need to know in the most fast moving daily program in America. Ben brutally breaks down the culture and never gives an inch! Monday thru Friday."]

# model.sample
prefix_allowed_tokens_fn = get_prefix_allowed_tokens_fn(model, sentences)
out = model.sample(
    sentences,
    prefix_allowed_tokens_fn=prefix_allowed_tokens_fn,
)
print(out)

# get_entity_spans
entity_spans = get_entity_spans( model, sentences)
print(get_markdown(sentences, entity_spans))