Sean-Case commited on
Commit
9a3229c
1 Parent(s): 8fe4ba4

Added nltk wordnet download to chatfuncs

Browse files
Files changed (1) hide show
  1. chatfuncs/chatfuncs.py +2 -0
chatfuncs/chatfuncs.py CHANGED
@@ -26,10 +26,12 @@ from langchain.docstore.document import Document
26
 
27
  # For keyword extraction
28
  import nltk
 
29
  from nltk.corpus import stopwords
30
  from nltk.tokenize import RegexpTokenizer
31
  from nltk.stem import WordNetLemmatizer
32
  import keybert
 
33
  #from transformers.pipelines import pipeline
34
 
35
  # For Name Entity Recognition model
 
26
 
27
  # For keyword extraction
28
  import nltk
29
+ nltk.download('wordnet')
30
  from nltk.corpus import stopwords
31
  from nltk.tokenize import RegexpTokenizer
32
  from nltk.stem import WordNetLemmatizer
33
  import keybert
34
+
35
  #from transformers.pipelines import pipeline
36
 
37
  # For Name Entity Recognition model