DD8943 commited on
Commit
cfd1d13
·
verified ·
1 Parent(s): 63c936e

Update file

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -9,11 +9,14 @@ import os
9
  import joblib
10
  from openai import OpenAI
11
 
 
 
 
 
 
 
12
  client = OpenAI() # Uses env var OPENAI_API_KEY
13
- os.environ["HF_HOME"] = "/app/hf_cache"
14
- os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
15
- os.environ["HF_HUB_CACHE"] = "/app/hf_cache"
16
- os.environ["TORCH_HOME"] = "/app/hf_cache"
17
 
18
 
19
  @st.cache_data
 
9
  import joblib
10
  from openai import OpenAI
11
 
12
+ import os
13
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
14
+ os.environ["HF_HOME"] = "/tmp"
15
+ os.environ["XDG_CACHE_HOME"] = "/tmp"
16
+ os.environ["SENTENCE_TRANSFORMERS_HOME"] = "/tmp"
17
+ os.environ["STREAMLIT_HOME"] = "/tmp"
18
  client = OpenAI() # Uses env var OPENAI_API_KEY
19
+
 
 
 
20
 
21
 
22
  @st.cache_data