Update pages/type_text.py

pages/type_text.py

@@ -44,19 +44,14 @@ numMAPPINGS_input = 5
 
 @st.cache_resource
 def load_model():
-    #st.markdown("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') 
 #model = SentenceTransformer('clips/mfaq')
-load_model()
+#load_model()
 
 INTdesc_embedding = model.encode(INTdesc_input)
 
@@ -112,7 +107,7 @@ if INTdesc_input is not None and createSBScodes_clicked == True:
             col1.write("%.4f" % result[3]["score"])
             col2.write(df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[3]["corpus_id"]],"SBS_Code_Hyphenated"].values[0])
             col3.write(SBScorpus[result[3]["corpus_id"]])
-            dictA["Score"].append("%.4f" % result[3]["score"]), dictA["SBS Code"].append(df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[3]["corpus_id"]],"SBS_Code_Hyphenated"].values[0]), dictA["SBS Description V2.0"].append(SBScorpus[result[3]["corpus_id"]])
+            dictA["Score"].append("%.4f" % result[3]["score"]), dictA["SBS Code"].append(df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[3]["corpus_id"]],"SBS_Code_Hyphenated"].values[0]), # dictA["SBS Description V2.0"].append(SBScorpus[result[3]["corpus_id"]])
             
             col1.write("%.4f" % result[4]["score"])
             col2.write(df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[4]["corpus_id"]],"SBS_Code_Hyphenated"].values[0])