Update functions.py

functions.py

@@ -38,10 +38,15 @@ def load_models():
     sent_pipe = pipeline("text-classification",model=q_model, tokenizer=q_tokenizer)
     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)
-    sbert = SentenceTransformer("all-mpnet-base-v2")
     cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-12-v2')
     
-    return asr_model, sent_pipe, sum_pipe, ner_pipe, sbert, cross_encoder
+    return asr_model, sent_pipe, sum_pipe, ner_pipe, cross_encoder
+    
+@st.experimental_singleton(suppress_st_warning=True)    
+def load_sbert(model_name):
+    sbert = SentenceTransformer(model_name)
+    
+    return sbert
 
 @st.experimental_singleton(suppress_st_warning=True)
 def get_spacy():
@@ -306,4 +311,4 @@ def fin_ext(text):
     return make_spans(text,results)
     
 nlp = get_spacy()    
-asr_model, sent_pipe, sum_pipe, ner_pipe, sbert, cross_encoder  = load_models()
+asr_model, sent_pipe, sum_pipe, ner_pipe, cross_encoder  = load_models()