Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
@@ -7,14 +7,13 @@ from langdetect import detect, DetectorFactory
|
|
7 |
# Ensure consistent language detection results
|
8 |
DetectorFactory.seed = 0
|
9 |
|
10 |
-
#
|
11 |
-
os.environ["HF_HOME"] = "/
|
12 |
-
os.environ["TRANSFORMERS_CACHE"] = "/
|
13 |
|
14 |
-
# Create cache directory
|
15 |
cache_dir = os.environ["HF_HOME"]
|
16 |
os.makedirs(cache_dir, exist_ok=True)
|
17 |
-
os.chmod(cache_dir, 0o755) # Set read/write/execute permissions for owner
|
18 |
|
19 |
# Retrieve Hugging Face token from environment variable
|
20 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
7 |
# Ensure consistent language detection results
|
8 |
DetectorFactory.seed = 0
|
9 |
|
10 |
+
# Use /tmp for Hugging Face cache to avoid permission errors
|
11 |
+
os.environ["HF_HOME"] = "/tmp/huggingface_cache"
|
12 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
|
13 |
|
14 |
+
# Create cache directory if it doesn't exist
|
15 |
cache_dir = os.environ["HF_HOME"]
|
16 |
os.makedirs(cache_dir, exist_ok=True)
|
|
|
17 |
|
18 |
# Retrieve Hugging Face token from environment variable
|
19 |
HF_TOKEN = os.getenv("HF_TOKEN")
|