Update functions.py
Browse files- functions.py +1 -1
functions.py
CHANGED
@@ -24,7 +24,7 @@ def load_models():
|
|
24 |
ner_tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large-finetuned-conll03-english")
|
25 |
sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
|
26 |
sum_pipe = pipeline("summarization",model="facebook/bart-large-cnn", tokenizer="facebook/bart-large-cnn")
|
27 |
-
|
28 |
sbert = SentenceTransformer("all-mpnet-base-v2")
|
29 |
cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
|
30 |
|
|
|
24 |
ner_tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-large-finetuned-conll03-english")
|
25 |
sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
|
26 |
sum_pipe = pipeline("summarization",model="facebook/bart-large-cnn", tokenizer="facebook/bart-large-cnn")
|
27 |
+
ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
|
28 |
sbert = SentenceTransformer("all-mpnet-base-v2")
|
29 |
cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
|
30 |
|