Views
No views yet
1from tibert import BertForCoreferenceResolution, predict_coref_simple
2from tibert.utils import pprint_coreference_document
3from transformers import BertTokenizerFast
4
5model = BertForCoreferenceResolution.from_pretrained(
6 "compnet-renard/bert-base-cased-literary-coref"
7)
8tokenizer = BertTokenizerFast.from_pretrained("bert-base-cased")
9
10coref_out = predict_coref_simple(
11 "Sli did not want the earpods. He didn't like them.", model, tokenizer
12)
13
14pprint_coreference_document(coref_out)>>> (0 Sli ) did not want the earpods. (0 He ) didn't like them.