Update app.py

app.py

@@ -7,9 +7,21 @@ from transformers import BertTokenizer, TFBertForSequenceClassification
 import logging
 import random
 import time
+import os
 
-nltk.download('punkt')
-nltk.download('wordnet')
+# Define a function to download NLTK data if it doesn't already exist
+def download_nltk_data():
+    nltk_data_dir = os.path.join(os.path.expanduser("~"), "nltk_data")
+    if not os.path.exists(nltk_data_dir):
+        os.makedirs(nltk_data_dir)
+
+    if not os.path.exists(os.path.join(nltk_data_dir, "tokenizers", "punkt")):
+        nltk.download('punkt', download_dir=nltk_data_dir)
+    
+    if not os.path.exists(os.path.join(nltk_data_dir, "corpora", "wordnet")):
+        nltk.download('wordnet', download_dir=nltk_data_dir)
+
+download_nltk_data()
 
 app = Flask(__name__)