Update citekit/cite_modules/augment_model.py
Browse files
citekit/cite_modules/augment_model.py
CHANGED
|
@@ -30,7 +30,6 @@ class Retriever(Module):
|
|
| 30 |
self.bm25_module_loaded = False
|
| 31 |
from rank_bm25 import BM25Okapi
|
| 32 |
import nltk
|
| 33 |
-
nltk.download('punkt')
|
| 34 |
from nltk.tokenize import word_tokenize
|
| 35 |
self.word_tokenize = word_tokenize
|
| 36 |
self.BM25Okapi = BM25Okapi
|
|
|
|
| 30 |
self.bm25_module_loaded = False
|
| 31 |
from rank_bm25 import BM25Okapi
|
| 32 |
import nltk
|
|
|
|
| 33 |
from nltk.tokenize import word_tokenize
|
| 34 |
self.word_tokenize = word_tokenize
|
| 35 |
self.BM25Okapi = BM25Okapi
|