import time
from kpe import KPE
import sys
import os


if __name__ == '__main__':
    TRAINED_MODEL_ADDR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'trained_model_10000.pt')
    text_addr = sys.argv[1]
    text = open(text_addr).read()
    kpe = KPE(trained_kpe_model= TRAINED_MODEL_ADDR, flair_ner_model='flair/ner-english-ontonotes-large', device='cpu')
    s =time.time()
    print(kpe.extract(text))
    print(time.time() - s)