nickmuchi commited on
Commit
7e12771
1 Parent(s): 8310667

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -42,7 +42,7 @@ def load_models():
42
  sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
43
  sum_pipe = pipeline("summarization",model="facebook/bart-large-cnn", tokenizer="facebook/bart-large-cnn",clean_up_tokenization_spaces=True)
44
  ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
45
- cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
46
 
47
  return sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer
48
 
 
42
  sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
43
  sum_pipe = pipeline("summarization",model="facebook/bart-large-cnn", tokenizer="facebook/bart-large-cnn",clean_up_tokenization_spaces=True)
44
  ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
45
+ cross_encoder = CrossEncoder('cross-encoder/mmarco-mMiniLMv2-L12-H384-v1') #cross-encoder/ms-marco-MiniLM-L-12-v2
46
 
47
  return sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer
48