Satyam0077 commited on
Commit
d31dcb9
·
verified ·
1 Parent(s): fdf044d

Update src/inference.py

Browse files
Files changed (1) hide show
  1. 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 to avoid runtime errors
9
- nltk.download('punkt')
 
 
 
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