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