JaySenpai commited on
Commit
5d9b172
·
verified ·
1 Parent(s): d41e2a7

Update app/main.py

Browse files

Fix cache directory permissions for Hugging Face model

Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
 
3
  # Set a writable cache directory inside the Space container
4
- os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"
5
 
6
 
7
  from fastapi import FastAPI
 
1
  import os
2
 
3
  # Set a writable cache directory inside the Space container
4
+ os.environ["TRANSFORMERS_CACHE"] = "/code/cache"
5
 
6
 
7
  from fastapi import FastAPI