Sean-Case
commited on
Commit
•
9a3229c
1
Parent(s):
8fe4ba4
Added nltk wordnet download to chatfuncs
Browse files- 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
|