Update main.py

main.py

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