Flair_NER/test_inf.py

11 lines
545 B
Python

from ner_proccess import inference_main
from funcs import save_to_file_by_address, read_file_by_address
import os
address2 = os.getcwd() + '/Flair_NER/data/ner_reg_list.txt'
text = read_file_by_address(address2)
id_lines = text.splitlines()
section = '''وزارت دارائی مكلف است پس از افتتاح مجلسین مجوز قانونی این تصویبنامه را تحصیل نماید.'''
ner_obj_list, content_ai, result = inference_main('orgcatorg/xlm-v-base-ner', section)
if not result[0]:
print(result)
print('ok')