Update functions.py
Browse files- functions.py +1 -1
functions.py
CHANGED
@@ -83,7 +83,7 @@ def load_models():
|
|
83 |
sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
|
84 |
sum_pipe = pipeline("summarization",model="philschmid/flan-t5-base-samsum",clean_up_tokenization_spaces=True)
|
85 |
ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
|
86 |
-
cross_encoder = CrossEncoder('cross-encoder/
|
87 |
sbert = SentenceTransformer('all-MiniLM-L6-v2')
|
88 |
|
89 |
return sent_pipe, sum_pipe, ner_pipe, cross_encoder, sbert
|
|
|
83 |
sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
|
84 |
sum_pipe = pipeline("summarization",model="philschmid/flan-t5-base-samsum",clean_up_tokenization_spaces=True)
|
85 |
ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
|
86 |
+
cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2') #cross-encoder/ms-marco-MiniLM-L-12-v2
|
87 |
sbert = SentenceTransformer('all-MiniLM-L6-v2')
|
88 |
|
89 |
return sent_pipe, sum_pipe, ner_pipe, cross_encoder, sbert
|