16pramodh commited on
Commit
f4d4584
·
1 Parent(s): 33abc82

resolve caching issue

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,5 +1,7 @@
1
  import os
2
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/cache" # MUST be before HF imports
 
 
3
 
4
  from fastapi import FastAPI
5
  from pydantic import BaseModel
 
1
  import os
2
+ os.environ['HF_HOME'] = '/tmp/hf_home'
3
+ os.environ['HF_DATASETS_CACHE'] = '/tmp/hf_datasets_cache'
4
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/cache'
5
 
6
  from fastapi import FastAPI
7
  from pydantic import BaseModel