ALLOUNE commited on
Commit
dd1567e
·
1 Parent(s): ef52d9f
Files changed (2) hide show
  1. Dockerfile +3 -0
  2. src/streamlit_app.py +2 -0
Dockerfile CHANGED
@@ -12,6 +12,9 @@ COPY requirements.txt ./
12
  COPY src/ ./src/
13
  COPY SAOKE_technologies.xlsx ./
14
 
 
 
 
15
  RUN pip3 install -r requirements.txt
16
 
17
  EXPOSE 8501
 
12
  COPY src/ ./src/
13
  COPY SAOKE_technologies.xlsx ./
14
 
15
+ RUN mkdir -p /app/hf_cache
16
+ ENV HF_HOME=/app/hf_cache
17
+
18
  RUN pip3 install -r requirements.txt
19
 
20
  EXPOSE 8501
src/streamlit_app.py CHANGED
@@ -9,6 +9,8 @@ from google.genai import Client, types
9
  from typing import List, Dict
10
  import os
11
 
 
 
12
  # Page configuration
13
  st.set_page_config(
14
  page_title="SAOKE Problem Solver",
 
9
  from typing import List, Dict
10
  import os
11
 
12
+ os.environ["HF_HOME"] = "/app/hf_cache"
13
+
14
  # Page configuration
15
  st.set_page_config(
16
  page_title="SAOKE Problem Solver",