georad commited on
Commit
b4977be
·
verified ·
1 Parent(s): b572574

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -2
pages/type_text.py CHANGED
@@ -44,10 +44,10 @@ numMAPPINGS_input = 5
44
 
45
  @st.cache_resource
46
  def load_model():
47
- st.header("Sentence Transformer")
48
  model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
49
  st.success("Loaded model!")
50
- model.encode(INTdesc_input)
51
  #st.write("Turning on evaluation mode...")
52
  #model.eval()
53
  #st.write("Here's the model:")
 
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!")
50
+ INTdesc_embedding = model.encode(INTdesc_input)
51
  #st.write("Turning on evaluation mode...")
52
  #model.eval()
53
  #st.write("Here's the model:")