Spaces:
Runtime error
Runtime error
fix internal run issue
Browse files- src/streamlit_app.py +6 -4
src/streamlit_app.py
CHANGED
|
@@ -1,8 +1,10 @@
|
|
| 1 |
import os
|
| 2 |
-
|
| 3 |
-
os.
|
| 4 |
-
os.environ['TRANSFORMERS_CACHE'] =
|
| 5 |
-
os.environ['
|
|
|
|
|
|
|
| 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
|