Spaces:
Sleeping
Sleeping
Ezi Ozoani
commited on
Commit
·
13af81f
1
Parent(s):
967976f
maybe this fixes it
Browse files- read_extract.py +2 -1
read_extract.py
CHANGED
@@ -224,7 +224,8 @@ def get_keywords(datatype, task, field, pos_text, neg_text):
|
|
224 |
print('Extracting keywords...')
|
225 |
datatype, task, field = datatype.lower(), task.lower(), field.lower()
|
226 |
|
227 |
-
stop_words = set(stopwords.words('english'))
|
|
|
228 |
stop_words = stop_words.union({'license', 'licensing', 'licensor', 'copyright', 'copyrights', 'patent'})
|
229 |
|
230 |
pos_kw_model = KeyBERT()
|
|
|
224 |
print('Extracting keywords...')
|
225 |
datatype, task, field = datatype.lower(), task.lower(), field.lower()
|
226 |
|
227 |
+
#stop_words = set(stopwords.words('english'))
|
228 |
+
stop_words = nltk.corpus.stopwords.words('english')
|
229 |
stop_words = stop_words.union({'license', 'licensing', 'licensor', 'copyright', 'copyrights', 'patent'})
|
230 |
|
231 |
pos_kw_model = KeyBERT()
|