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")
|