Update pages/type_text.py

pages/type_text.py

@@ -47,10 +47,11 @@ def load_model():
     st.header("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 model
+    return INTdesc_embedding
 
 #model = SentenceTransformer('all-MiniLM-L6-v2') # fastest
 #model = SentenceTransformer('all-mpnet-base-v2') # best performance
@@ -58,7 +59,7 @@ def load_model():
 #model = SentenceTransformer('sentence-transformers/msmarco-bert-base-dot-v5') 
 #model = SentenceTransformer('clips/mfaq')
 
-INTdesc_embedding = model.encode(INTdesc_input)
+#INTdesc_embedding = model.encode(INTdesc_input)
 
 # Semantic search, Compute cosine similarity between all pairs of SBS descriptions