Update pages/type_text.py
Browse files- pages/type_text.py +2 -2
pages/type_text.py
CHANGED
@@ -43,7 +43,7 @@ numMAPPINGS_input = 5
|
|
43 |
#st.button("Clear text", on_click=on_click)
|
44 |
|
45 |
@st.cache_resource
|
46 |
-
def load_model(
|
47 |
st.markdown("Sentence Transformer")
|
48 |
model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
|
49 |
st.success("Loaded model!")
|
@@ -59,7 +59,7 @@ def load_model(INTdesc_input):
|
|
59 |
#model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5')
|
60 |
#model = SentenceTransformer('clips/mfaq')
|
61 |
|
62 |
-
|
63 |
#INTdesc_embedding = model.encode(INTdesc_input)
|
64 |
|
65 |
# Semantic search, Compute cosine similarity between all pairs of SBS descriptions
|
|
|
43 |
#st.button("Clear text", on_click=on_click)
|
44 |
|
45 |
@st.cache_resource
|
46 |
+
def load_model():
|
47 |
st.markdown("Sentence Transformer")
|
48 |
model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
|
49 |
st.success("Loaded model!")
|
|
|
59 |
#model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5')
|
60 |
#model = SentenceTransformer('clips/mfaq')
|
61 |
|
62 |
+
INTdesc_embedding = load_model(INTdesc_input)
|
63 |
#INTdesc_embedding = model.encode(INTdesc_input)
|
64 |
|
65 |
# Semantic search, Compute cosine similarity between all pairs of SBS descriptions
|