shubhamgs commited on
Commit
0d2935b
·
verified ·
1 Parent(s): 2f9f464

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -12,10 +12,16 @@ RUN apt-get update && apt-get install -y \
12
  COPY requirements.txt ./
13
  COPY src/ ./src/
14
 
 
15
  RUN pip3 install --upgrade -r requirements.txt
16
 
17
  EXPOSE 8501
18
 
 
 
 
 
 
19
  HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
20
 
21
  ENTRYPOINT ["streamlit", "run", "src/tinny_llama.py", "--server.port=8501", "--server.address=0.0.0.0"]
 
12
  COPY requirements.txt ./
13
  COPY src/ ./src/
14
 
15
+ RUN pip install --upgrade pip
16
  RUN pip3 install --upgrade -r requirements.txt
17
 
18
  EXPOSE 8501
19
 
20
+ ENV STREAMLIT_CONFIG_DIR=/tmp/.streamlit
21
+ ENV HF_HOME=/tmp/hf_cache
22
+ ENV TRANSFORMERS_CACHE=/tmp/hf_cache
23
+ ENV SENTENCE_TRANSFORMERS_HOME=/tmp/st_cache
24
+
25
  HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
26
 
27
  ENTRYPOINT ["streamlit", "run", "src/tinny_llama.py", "--server.port=8501", "--server.address=0.0.0.0"]