ink85 commited on
Commit
a701d3d
·
verified ·
1 Parent(s): cdfaada

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -1,5 +1,16 @@
1
- import streamlit as st
2
  import os
 
 
 
 
 
 
 
 
 
 
 
 
3
  import json
4
  import re
5
  import fitz # PyMuPDF
 
 
1
  import os
2
+
3
+ # Redirect Hugging Face cache to writable directory
4
+ os.environ["HF_HOME"] = "/app/.cache/huggingface"
5
+ os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/transformers"
6
+ os.environ["HF_HUB_CACHE"] = "/app/.cache/hub"
7
+
8
+ # Make sure the folders exist
9
+ os.makedirs(os.environ["HF_HOME"], exist_ok=True)
10
+ os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
11
+ os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
12
+
13
+ import streamlit as st
14
  import json
15
  import re
16
  import fitz # PyMuPDF