Update main.py

main.py

@@ -9,8 +9,12 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import os
 
-nltk.data.path.append('/app/nltk_data')
+nltk_data_path = "/app/nltk_data"
+os.makedirs(nltk_data_path, exist_ok=True)
+nltk.data.path.append(nltk_data_path)
+
 os.environ['TRANSFORMERS_CACHE'] = '/app/transformers_cache'
+os.makedirs('/app/transformers_cache', exist_ok=True) 
 
 nltk.download('punkt')
 nltk.download('stopwords')