Update pages/type_text.py
Browse files- 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 =
|
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)
|