KarthikAI commited on
Commit
4cf3c8d
·
verified ·
1 Parent(s): fb884af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
- os.environ['TRANSFORMERS_CACHE'] = '/workspace/cache/hf'
3
- os.environ['HF_HOME'] = '/workspace/cache/hf'
4
- os.makedirs('/workspace/cache/hf', exist_ok=True)
 
5
 
6
  from fastapi import FastAPI, UploadFile, File, Form
7
  from fastapi.responses import StreamingResponse
 
1
  import os
2
+ cache_dir = os.path.expanduser('~/.cache/hf')
3
+ os.environ['TRANSFORMERS_CACHE'] = cache_dir
4
+ os.environ['HF_HOME'] = cache_dir
5
+ os.makedirs(cache_dir, exist_ok=True)
6
 
7
  from fastapi import FastAPI, UploadFile, File, Form
8
  from fastapi.responses import StreamingResponse