Update functions.py

functions.py

@@ -112,8 +112,9 @@ def load_models():
     sum_pipe = pipeline("summarization",model="facebook/bart-large-cnn", tokenizer="facebook/bart-large-cnn",clean_up_tokenization_spaces=True)
     ner_pipe = pipeline("ner", model=ner_model, tokenizer=ner_tokenizer, grouped_entities=True)
     cross_encoder = CrossEncoder('cross-encoder/mmarco-mMiniLMv2-L12-H384-v1') #cross-encoder/ms-marco-MiniLM-L-12-v2
+    sbert = SentenceTransformer(all-MiniLM-L6-v2)
     
-    return sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer
+    return sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer, sbert
 
 @st.experimental_singleton(suppress_st_warning=True)
 def load_asr_model(asr_model_name):
@@ -752,8 +753,7 @@ def save_network_html(kb, filename="network.html"):
     )
     net.set_edge_smooth('dynamic')
     net.show(filename)
-            
 
 nlp = get_spacy()    
 
-sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer  = load_models()
+sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer, sbert  = load_models()