Update src/streamlit_app.py

src/streamlit_app.py

@@ -16,11 +16,18 @@ 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")
+# 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"
 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.download('punkt', download_dir=nltk_data_dir)
+nltk.download('averaged_perceptron_tagger', download_dir=nltk_data_dir)
+
 # Load models
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", device=torch.device("cpu"))
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad", device=torch.device("cpu"))