Yoonc commited on
Commit
2c9c7aa
·
verified ·
1 Parent(s): 5485161

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -0
Dockerfile CHANGED
@@ -39,5 +39,11 @@ ENV STREAMLIT_SERVER_PORT=7860
39
  ENV STREAMLIT_SERVER_ENABLECORS=false
40
  ENV STREAMLIT_SERVER_HEADLESS=true
41
 
 
 
 
 
 
 
42
  # Start the Streamlit app
43
  CMD ["streamlit", "run", "main.py"]
 
39
  ENV STREAMLIT_SERVER_ENABLECORS=false
40
  ENV STREAMLIT_SERVER_HEADLESS=true
41
 
42
+
43
+ ENV TRANSFORMERS_CACHE=/app/cache
44
+ ENV HF_HOME=/app/cache
45
+ RUN mkdir -p /app/cache
46
+
47
+
48
  # Start the Streamlit app
49
  CMD ["streamlit", "run", "main.py"]