Update src/streamlit_app.py

src/streamlit_app.py

@@ -9,11 +9,13 @@ from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.models import load_model
 import nltk
 
-nltk.data.path.append('src')
+# Use /tmp for NLTK data (writable in Hugging Face Spaces)
+nltk_data_dir = "/tmp/nltk_data"
+nltk.data.path.append(nltk_data_dir)
 
-# Download the stopwords resource
-nltk.download('stopwords', download_dir='src')
-nltk.download('punkt', download_dir='src')
+# Download the stopwords and punkt resources
+nltk.download('stopwords', download_dir=nltk_data_dir)
+nltk.download('punkt', download_dir=nltk_data_dir)
 
 # Load the trained model
 model = load_model('src/model.keras')
@@ -49,7 +51,7 @@ def text_preprocessing(text):
     tokens = word_tokenize(text)
 
     # Stopwords removal
-    # tokens = [word for word in tokens if word not in stpwds_id]
+    tokens = [word for word in tokens if word not in stpwds_id]
 
     # Combining Tokens
     text = ' '.join(tokens)