dejanseo commited on
Commit
fbf3acc
·
verified ·
1 Parent(s): 34def1e

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- os.environ["HF_HOME"] = "./hf_cache"
 
 
 
 
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")