Update main.py

main.py

@@ -9,17 +9,17 @@ from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
 import os
 
-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) 
+os.environ["HF_HOME"] = "/home/user/huggingface"
+nltk.data.path.append("/home/user/nltk_data")
 
-nltk.download('punkt', download_dir=nltk_data_path)
-nltk.download('stopwords', download_dir=nltk_data_path)
-nltk.download('averaged_perceptron_tagger', download_dir=nltk_data_path)
-nltk.download('punkt_tab', download_dir=nltk_data_path)
+os.makedirs("/home/user/huggingface", exist_ok=True)
+os.makedirs("/home/user/nltk_data", exist_ok=True)
+
+nltk.download('punkt', download_dir='/home/user/nltk_data')
+nltk.download('stopwords', download_dir='/home/user/nltk_data')
+nltk.download('averaged_perceptron_tagger', download_dir='/home/user/nltk_data')
+nltk.download('punkt_tab', download_dir='/home/user/nltk_data')
 
 
 # 1. Function for getting news via NewsAPI