Update pages/type_text.py

pages/type_text.py

@@ -47,11 +47,10 @@ def load_model(INTdesc):
     st.markdown("Sentence Transformer")
     model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
     st.success("Loaded model!")
-    INTdesc_embedding = model.encode(INTdesc_input)
     #st.write("Turning on evaluation mode...")
     #model.eval()
     #st.write("Here's the model:")
-    return INTdesc_embedding
+    return model
 
 #model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
 #model = SentenceTransformer('all-mpnet-base-v2') # best performance
@@ -59,8 +58,9 @@ def load_model(INTdesc):
 #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5') 
 #model = SentenceTransformer('clips/mfaq')
 
-INTdesc_embedding = load_model(INTdesc=INTdesc_input)
-#INTdesc_embedding = model.encode(INTdesc_input)
+load_model()
+#INTdesc_embedding = load_model(INTdesc=INTdesc_input)
+INTdesc_embedding = model.encode(INTdesc_input)
 
 # Semantic search, Compute cosine similarity between all pairs of SBS descriptions