compnet-renard/bert-base-cased-literary-coref

This model should be used with Tibert, a BERT-based end-to-end coreference solver.

Example code:

from tibert import BertForCoreferenceResolution, predict_coref_simple
from tibert.utils import pprint_coreference_document
from transformers import BertTokenizerFast

model = BertForCoreferenceResolution.from_pretrained(
    "compnet-renard/bert-base-cased-literary-coref"
)
tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")

coref_out = predict_coref_simple(
    "Sli did not want the earpods. He didn't like them.", model, tokenizer
)

pprint_coreference_document(coref_out)

>>> (0 Sli ) did not want the earpods. (0 He ) didn't like them.