Ehrii commited on
Commit
7602c5f
·
1 Parent(s): 1d29239

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -8,8 +8,8 @@ from langdetect import detect, DetectorFactory
8
  DetectorFactory.seed = 0
9
 
10
  # Set Hugging Face cache directory to a writable location
11
- os.environ["HF_HOME"] = "/tmp/huggingface"
12
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
13
 
14
  # Create cache directory with proper permissions
15
  cache_dir = os.environ["HF_HOME"]
 
8
  DetectorFactory.seed = 0
9
 
10
  # Set Hugging Face cache directory to a writable location
11
+ os.environ["HF_HOME"] = "/app/huggingface_cache"
12
+ os.environ["TRANSFORMERS_CACHE"] = "/app/huggingface_cache"
13
 
14
  # Create cache directory with proper permissions
15
  cache_dir = os.environ["HF_HOME"]