georad commited on
Commit
cd5f3f2
·
verified ·
1 Parent(s): 08afd6a

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -2
pages/type_text.py CHANGED
@@ -9,7 +9,7 @@ from sentence_transformers import SentenceTransformer, util
9
  import os
10
  os.getenv("HF_TOKEN")
11
 
12
- st.header("Map internal descriptions to SBS codes using Sentence Transformer + Reasoning Models")
13
  st.subheader("Select specific Chapter for quicker results")
14
  #df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows.csv")
15
  df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv")
@@ -171,7 +171,7 @@ dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
171
  dfALL = pd.DataFrame.from_dict(dictA)
172
 
173
 
174
- if INTdesc_input is not None and if st.button(...):
175
  #my_model_results = pipeline("ner", model= "checkpoint-92")
176
  HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
177
  HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
 
9
  import os
10
  os.getenv("HF_TOKEN")
11
 
12
+ st.header("Map internal descriptions to SBS codes using Sentence Transformer plus Reasoning Models")
13
  st.subheader("Select specific Chapter for quicker results")
14
  #df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows.csv")
15
  df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv")
 
171
  dfALL = pd.DataFrame.from_dict(dictA)
172
 
173
 
174
+ if INTdesc_input is not None and if st.button("user_clickedSBS"):
175
  #my_model_results = pipeline("ner", model= "checkpoint-92")
176
  HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
177
  HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)