nikos99n commited on
Commit
2be0d27
·
verified ·
1 Parent(s): 0ee89fd

fix internal run issue

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -4
src/streamlit_app.py CHANGED
@@ -1,8 +1,10 @@
1
  import os
2
- cache_dir = '/tmp/hf_cache'
3
- os.makedirs(cache_dir, exist_ok=True)
4
- os.environ['TRANSFORMERS_CACHE'] = cache_dir
5
- os.environ['HF_HOME'] = cache_dir
 
 
6
 
7
  from sentence_transformers import SentenceTransformer
8
  import streamlit as st
 
1
  import os
2
+
3
+ os.environ['HF_HOME'] = '/tmp/hf_cache'
4
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
5
+ os.environ['STREAMLIT_HOME'] = '/tmp/streamlit'
6
+ os.makedirs('/tmp/hf_cache', exist_ok=True)
7
+ os.makedirs('/tmp/streamlit', exist_ok=True)
8
 
9
  from sentence_transformers import SentenceTransformer
10
  import streamlit as st