Update pages/type_text.py

pages/type_text.py

@@ -103,6 +103,22 @@ numMAPPINGS_input = 5
 #numMAPPINGS_input = st.text_input("Type number of mappings and hit Enter", key="user_input_numMAPPINGS")
 #st.button("Clear text", on_click=on_click)
 
+SentenceTransformer = st.selectbox(
+    "Select the Sentence Transformer model",
+    ("all-MiniLM-L6-v2 (original, general domain)", "all-MiniLM-L6-v2 (fine-tuned, medical domain)", "all-mpnet-base-v2 (original, general domain)", "all-mpnet-base-v2 (fine-tuned, medical domain)"]
+)
+st.write("You selected:", SentenceTransformer)
+
+
+
+
+
+
+
+
+
+
+
 @st.cache_resource
 def load_model():
     model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
@@ -115,6 +131,18 @@ model = load_model()
 #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5') 
 #model = SentenceTransformer('clips/mfaq')
 
+
+
+
+
+
+
+
+
+
+
+
+
 INTdesc_embedding = model.encode(INTdesc_input)
 
 # Semantic search, Compute cosine similarity between INTdesc_embedding and SBS descriptions