georad commited on
Commit
c890c65
·
verified ·
1 Parent(s): 9a03d04

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +1 -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(INTdesc):
47
  st.markdown("Sentence Transformer")
48
  model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
49
  st.success("Loaded model!")
@@ -59,7 +59,6 @@ def load_model(INTdesc):
59
  #model = SentenceTransformer('clips/mfaq')
60
 
61
  load_model()
62
- #INTdesc_embedding = load_model(INTdesc=INTdesc_input)
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('clips/mfaq')
60
 
61
  load_model()
 
62
  INTdesc_embedding = model.encode(INTdesc_input)
63
 
64
  # Semantic search, Compute cosine similarity between all pairs of SBS descriptions