georad commited on
Commit
8b54142
·
verified ·
1 Parent(s): f660b63

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -2
pages/type_text.py CHANGED
@@ -17,7 +17,7 @@ device = get_device_map() # 'cpu'
17
 
18
  def on_click():
19
  st.session_state.user_input = ""
20
-
21
  #@st.cache
22
  def convert_df(df:pd.DataFrame):
23
  return df.to_csv(index=False).encode('utf-8')
@@ -72,7 +72,7 @@ df_SBS = pd.read_csv("SBS_V2_Table.csv", header=0, skip_blank_lines=False, skipr
72
  #st.write(df_SBS.head(5))
73
 
74
  SBScorpus = df_SBS['Long_Description'].values.tolist()
75
- SBScorpus_embeddings = INTdesc_embedding.encode(SBScorpus)
76
 
77
  #my_model_results = pipeline("ner", model= "checkpoint-92")
78
  HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
 
17
 
18
  def on_click():
19
  st.session_state.user_input = ""
20
+
21
  #@st.cache
22
  def convert_df(df:pd.DataFrame):
23
  return df.to_csv(index=False).encode('utf-8')
 
72
  #st.write(df_SBS.head(5))
73
 
74
  SBScorpus = df_SBS['Long_Description'].values.tolist()
75
+ SBScorpus_embeddings = model.encode(SBScorpus)
76
 
77
  #my_model_results = pipeline("ner", model= "checkpoint-92")
78
  HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)