Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,9 +16,15 @@ import random
 import torch  # Import torch
 
 # Download NLTK resources if not already downloaded
-nltk.download('averaged_perceptron_tagger', quiet=True)
-nltk.download('punkt', quiet=True)
-nltk.download('punkt_tab', quiet=True)
+# 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 = 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)
 
 # Load models
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=torch.device("cpu"))