Flair_NER/classifier.py

13 lines
331 B
Python

from transformers import pipeline
import os
address = os.getcwd()
modelAddress = "~/tnlp/jokar/Flair_NER/taggers/final-model.pt"
print(modelAddress)
print('*'*100)
print('*'*100)
print('*'*100)
classifier = pipeline('ner', model=modelAddress)
from inference import read_file
input_sentence = read_file()
classifier(input_sentence)