Spaces:
Runtime error
Runtime error
Update server.py
Browse files
server.py
CHANGED
@@ -6,11 +6,11 @@ import torch
|
|
6 |
|
7 |
app = FastAPI()
|
8 |
|
9 |
-
# β
Fix:
|
10 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp"
|
11 |
os.environ["HF_HOME"] = "/tmp"
|
12 |
|
13 |
-
# β
Ensure
|
14 |
if not os.path.exists("/tmp"):
|
15 |
os.makedirs("/tmp")
|
16 |
|
|
|
6 |
|
7 |
app = FastAPI()
|
8 |
|
9 |
+
# β
Fix: Set writable cache directory for Hugging Face models
|
10 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp"
|
11 |
os.environ["HF_HOME"] = "/tmp"
|
12 |
|
13 |
+
# β
Ensure cache directory exists
|
14 |
if not os.path.exists("/tmp"):
|
15 |
os.makedirs("/tmp")
|
16 |
|