Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,12 +16,7 @@ import random
 import torch  # Import torch
 
 # Download NLTK resources if not already downloaded
-# nltk_data_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), "nltk_data")
-# os.makedirs(nltk_data_dir, exist_ok=True)
-# nltk.download('averaged_perceptron_tagger', download_dir=nltk_data_dir, quiet=True)
-# nltk.data.path.append(nltk_data_dir)
-
-nltk_data_dir = "/tmp/nltk_data"
+nltk_data_dir = os.path.join(os.path.dirname(os.path.dirname(__file__)), "nltk_data")
 os.makedirs(nltk_data_dir, exist_ok=True)
 nltk.download('averaged_perceptron_tagger', download_dir=nltk_data_dir, quiet=True)
 nltk.data.path.append(nltk_data_dir)