kparkhade commited on
Commit
0af902d
·
verified ·
1 Parent(s): 5ad88ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,8 @@ def summarize_text(text):
34
 
35
  # Function: Keyword Extraction
36
  def extract_keywords(text):
37
- return models["keyword_extraction"].extract_keywords(text, ngram_range=(1, 2), stop_words='english')
 
38
 
39
  # Function: Named Entity Recognition (NER)
40
  def analyze_ner(text):
 
34
 
35
  # Function: Keyword Extraction
36
  def extract_keywords(text):
37
+ vectorizer = CountVectorizer(ngram_range=(1, 2))
38
+ return models["keyword_extraction"].extract_keywords(text, vectorizer=vectorizer, stop_words='english')
39
 
40
  # Function: Named Entity Recognition (NER)
41
  def analyze_ner(text):