rachman commited on
Commit
46e8a7c
·
verified ·
1 Parent(s): d85ef35

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +4 -2
src/streamlit_app.py CHANGED
@@ -9,9 +9,11 @@ from tensorflow.keras.preprocessing.sequence import pad_sequences
9
  from tensorflow.keras.models import load_model
10
  import nltk
11
 
 
 
12
  # Download the stopwords resource
13
- # nltk.download('stopwords')
14
- # nltk.download('punkt')
15
 
16
  # Load the trained model
17
  model = load_model('src/model.keras')
 
9
  from tensorflow.keras.models import load_model
10
  import nltk
11
 
12
+ nltk.data.path.append('src')
13
+
14
  # Download the stopwords resource
15
+ nltk.download('stopwords', download_dir='src')
16
+ nltk.download('punkt', download_dir='src')
17
 
18
  # Load the trained model
19
  model = load_model('src/model.keras')