Update src/inference.py
Browse files- src/inference.py +5 -2
src/inference.py
CHANGED
|
@@ -5,8 +5,11 @@ import scipy.sparse
|
|
| 5 |
from textblob import TextBlob
|
| 6 |
import nltk
|
| 7 |
|
| 8 |
-
# Download NLTK punkt tokenizer
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
from src.preprocessing import clean_text
|
| 12 |
from src.features import create_features
|
|
|
|
| 5 |
from textblob import TextBlob
|
| 6 |
import nltk
|
| 7 |
|
| 8 |
+
# Download NLTK punkt tokenizer if not already present
|
| 9 |
+
try:
|
| 10 |
+
nltk.data.find('tokenizers/punkt')
|
| 11 |
+
except LookupError:
|
| 12 |
+
nltk.download('punkt')
|
| 13 |
|
| 14 |
from src.preprocessing import clean_text
|
| 15 |
from src.features import create_features
|