mjpsm commited on
Commit
e0d585f
·
verified ·
1 Parent(s): 5ddb4bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -4,10 +4,17 @@ from fastapi import FastAPI
4
  from pydantic import BaseModel
5
  import torch
6
 
7
- # ✅ Fix for Hugging Face Spaces permission issue
8
- os.environ["HF_HOME"] = "/tmp/hf_cache"
9
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
10
- os.makedirs("/tmp/hf_cache", exist_ok=True)
 
 
 
 
 
 
 
11
 
12
  # Initialize FastAPI
13
  app = FastAPI(title="Check-ins Classifier API", version="1.0")
 
4
  from pydantic import BaseModel
5
  import torch
6
 
7
+ import os
8
+
9
+ # 🧱 Set all possible cache directories to writable locations
10
+ os.environ["HF_HOME"] = "/tmp/huggingface"
11
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface/transformers"
12
+ os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface/datasets"
13
+ os.environ["XDG_CACHE_HOME"] = "/tmp/huggingface" # prevents /.cache access
14
+
15
+ # Ensure directory exists
16
+ os.makedirs("/tmp/huggingface/transformers", exist_ok=True)
17
+
18
 
19
  # Initialize FastAPI
20
  app = FastAPI(title="Check-ins Classifier API", version="1.0")