Update pages/type_text.py

pages/type_text.py

@@ -42,8 +42,17 @@ numMAPPINGS_input = 5
 #numMAPPINGS_input = st.text_input("Type number of mappings and hit Enter", key="user_input_numMAPPINGS")
 #st.button("Clear text", on_click=on_click)
 
-
-model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
+@st.cache_resource
+def load_model():
+    st.header("Sentence Transformer")
+    model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
+    st.success("Loaded model!")
+    #st.write("Turning on evaluation mode...")
+    #model.eval()
+    #st.write("Here's the model:")
+    return model
+
+#model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
 #model = SentenceTransformer('all-mpnet-base-v2') # best performance
 #model = SentenceTransformers('all-distilroberta-v1')
 #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5') 
@@ -63,7 +72,7 @@ df_SBS = pd.read_csv("SBS_V2_Table.csv", header=0, skip_blank_lines=False, skipr
 #st.write(df_SBS.head(5))
 
 SBScorpus = df_SBS['Long_Description'].values.tolist()
-SBScorpus_embeddings = model.encode(SBScorpus)
+SBScorpus_embeddings = load_model().encode(SBScorpus)
 
 #my_model_results = pipeline("ner", model= "checkpoint-92")
 HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)