Update pages/type_text.py
Browse files- 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 |
-
|
146 |
-
|
147 |
-
|
148 |
-
|
|
|
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 = {
|