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

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -3
pages/type_text.py CHANGED
@@ -51,14 +51,13 @@ def load_model():
51
  #model.eval()
52
  #st.write("Here's the model:")
53
  return model
54
-
55
- #model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
56
  #model = SentenceTransformer('all-mpnet-base-v2') # best performance
57
  #model = SentenceTransformers('all-distilroberta-v1')
58
  #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5')
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
 
51
  #model.eval()
52
  #st.write("Here's the model:")
53
  return model
54
+ model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
 
55
  #model = SentenceTransformer('all-mpnet-base-v2') # best performance
56
  #model = SentenceTransformers('all-distilroberta-v1')
57
  #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5')
58
  #model = SentenceTransformer('clips/mfaq')
 
59
  load_model()
60
+
61
  INTdesc_embedding = model.encode(INTdesc_input)
62
 
63
  # Semantic search, Compute cosine similarity between all pairs of SBS descriptions