georad commited on
Commit
8397c5f
·
verified ·
1 Parent(s): 043184f

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +5 -4
pages/type_text.py CHANGED
@@ -142,10 +142,11 @@ INTdesc_embedding = model.encode(INTdesc_input)
142
  SBScorpus_embeddings = model.encode(SBScorpus)
143
 
144
  #if len(chapter_rows_indexes_list) >1:
145
- #my_model_results = pipeline("ner", model= "checkpoint-92")
146
- HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
147
- HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
148
- HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
 
149
 
150
  ## Define the Reasoning models
151
  rs_models = {
 
142
  SBScorpus_embeddings = model.encode(SBScorpus)
143
 
144
  #if len(chapter_rows_indexes_list) >1:
145
+ if INTdesc_input is not None:
146
+ #my_model_results = pipeline("ner", model= "checkpoint-92")
147
+ HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
148
+ HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
149
+ HF_model_results_displayed = HF_model_results_sorted[0:numMAPPINGS_input]
150
 
151
  ## Define the Reasoning models
152
  rs_models = {