Update src/streamlit_app.py
Browse files- src/streamlit_app.py +5 -1
src/streamlit_app.py
CHANGED
|
@@ -6,7 +6,11 @@ import numpy as np
|
|
| 6 |
import pandas as pd
|
| 7 |
import plotly.express as px
|
| 8 |
import os
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
# App title
|
| 12 |
st.title("Keyword Cosine Similarity Tool")
|
|
|
|
| 6 |
import pandas as pd
|
| 7 |
import plotly.express as px
|
| 8 |
import os
|
| 9 |
+
# Route all cache and config writes to Docker-writable dirs
|
| 10 |
+
os.environ["HF_HOME"] = "/app/hf_cache"
|
| 11 |
+
os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
|
| 12 |
+
os.environ["XDG_CACHE_HOME"] = "/app/hf_cache"
|
| 13 |
+
os.environ["STREAMLIT_HOME"] = "/app/.streamlit"
|
| 14 |
|
| 15 |
# App title
|
| 16 |
st.title("Keyword Cosine Similarity Tool")
|