Add application file

app.py

@@ -16,10 +16,23 @@ def main():
     logging.basicConfig(level=logging.INFO)
     logger = logging.getLogger(__name__)
     logger.info("Initializing training process...")
+    import os
     import nltk
+    
+    # Set up NLTK data directory
+    nltk_data_path = "./nltk_data"
+    os.makedirs(nltk_data_path, exist_ok=True)
+    nltk.data.path.append(nltk_data_path)
+
     # Ensure NLTK resources are available
-    nltk.download('punkt', quiet=True)
-    nltk.download('stopwords', quiet=True)
+    nltk.download('punkt', download_dir=nltk_data_path, quiet=True)
+    nltk.download('stopwords', download_dir=nltk_data_path, quiet=True)
+    
+    # Set up logging
+    logging.basicConfig(level=logging.INFO)
+    logger = logging.getLogger(__name__)
+    logger.info("Initializing training process...")
+    
     # Set a random seed for reproducibility
     set_seed(42)