sayef / bert-coref-resolver Goto Github PK

git clone --recurse-submodules https://github.com/sayef/bert-coref-resolver.git
cd bert-coref-resolver/coref
pip install -r requirements.txt
./setup_all.sh
./download_pretrained.sh spanbert_base models

import os
import nltk
nltk.download('punkt')

from resolver import Resolver
    

genre = "nw" # Other options: https://natural-language-understanding.fandom.com/wiki/OntoNotes
model_name = "spanbert_base" # The fine-tuned model to use. Options are: bert_base, spanbert_base, bert_large, spanbert_large
model_dir = "./coref/models" # where you have extracted pretrained model, from outside this module use absolute path

# coref resolver that loads model
coref_resolver = Resolver(genre, model_dir, model_name)
    
# text fot coref resolution
text = "Deepika has a dog. She loves him. The movie star has always been fond of animals."

# call resolve method as many without loading model again


resolved = coref_resolver.resolve(text)

print(resolved)