DevBM commited on
Commit
c9ff03b
1 Parent(s): ac0aab4

Remove enhanced_ner_entities while returning the keywords

Browse files
Files changed (1) hide show
  1. 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)