Rsnarsna commited on
Commit
9ce5ba0
·
verified ·
1 Parent(s): 48ea472

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -4,8 +4,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
4
  import torch
5
 
6
  # Set writable cache directories for Hugging Face Spaces
7
- os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
8
- os.environ["HF_HOME"] = "/app/.cache"
9
 
10
  app = FastAPI()
11
 
 
4
  import torch
5
 
6
  # Set writable cache directories for Hugging Face Spaces
7
+ #os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
8
+ #os.environ["HF_HOME"] = "/app/.cache"
9
 
10
  app = FastAPI()
11