Solved by nielsr in post #2 
              
              
                Hi, 
I suggest to take a look at this repo: GitHub - mandarjoshi90/coref: BERT for Coreference Resolution 
It includes multiple models (BERT, SpanBERT) fine-tuned on OntoNotes, an important benchmark for coreference resolution. 
There’s also a demo notebook, showcasing how to run inference for a new…