Update pages/type_text.py

pages/type_text.py

@@ -116,14 +116,15 @@ model = load_model()
 
 INTdesc_embedding = model.encode(INTdesc_input)
 
-# Semantic search, Compute cosine similarity between all pairs of SBS descriptions
+# Semantic search, Compute cosine similarity between INTdesc_embedding and SBS descriptions
 
 SBScorpus_embeddings = model.encode(SBScorpus)
 
-#my_model_results = pipeline("ner", model= "checkpoint-92")
-HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
-HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
-HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
+if len(chapter_rows_indexes_list) >1:
+    #my_model_results = pipeline("ner", model= "checkpoint-92")
+    HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
+    HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
+    HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
 
 @st.cache_resource
 def load_pipe():