diff --git a/func/ner/ner.py b/func/ner/ner.py index 57ec1992196f6b93944a71c6bd64d78f3096ec4f..7777b73f4e34eeed00a97105e3d9886330b98ac1 100644 --- a/func/ner/ner.py +++ b/func/ner/ner.py @@ -2,6 +2,7 @@ import torch from ckip_transformers.nlp import CkipWordSegmenter, CkipPosTagger, CkipNerChunker from transformers import pipeline + # Perform NER on Text def run_ner_on_text(page): ner_driver = CkipNerChunker(model="bert-base")