Spaces:
Sleeping
Sleeping
Remove enhanced_ner_entities while returning the keywords
Browse files- keyword_extraction.py +2 -1
keyword_extraction.py
CHANGED
@@ -109,7 +109,8 @@ def extract_keywords(text, extract_all):
|
|
109 |
print(f"\n\nSpacy Entities: {spacy_keywords} \n\n")
|
110 |
|
111 |
if extract_all is False:
|
112 |
-
combined_keywords_without_all = list(spacy_keywords.union(gliner_keywords).union(enhanced_ner_entities))
|
|
|
113 |
filtered_results = filter_keywords(combined_keywords_without_all)
|
114 |
print("Keywords returned: ",filtered_results)
|
115 |
return list(filtered_results)
|
|
|
109 |
print(f"\n\nSpacy Entities: {spacy_keywords} \n\n")
|
110 |
|
111 |
if extract_all is False:
|
112 |
+
# combined_keywords_without_all = list(spacy_keywords.union(gliner_keywords).union(enhanced_ner_entities))
|
113 |
+
combined_keywords_without_all = list(spacy_keywords.union(gliner_keywords))
|
114 |
filtered_results = filter_keywords(combined_keywords_without_all)
|
115 |
print("Keywords returned: ",filtered_results)
|
116 |
return list(filtered_results)
|