nam pham commited on
Commit
ba202aa
·
1 Parent(s): ae7776c

feat: update dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -0
Dockerfile CHANGED
@@ -1,3 +1,12 @@
1
  FROM nampham11062002/chatbot-api
2
 
 
 
 
 
 
 
 
 
 
3
  CMD ["/app/.venv/bin/python", "-m", "src.infastructure.api"]
 
1
  FROM nampham11062002/chatbot-api
2
 
3
+ # Create cache directory and set permissions
4
+ RUN mkdir -p /app/.cache/huggingface/hub && \
5
+ chmod -R 777 /app/.cache
6
+
7
+ # Set environment variables for Hugging Face cache
8
+ ENV HF_HOME=/app/.cache/huggingface
9
+ ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
10
+ ENV HF_DATASETS_CACHE=/app/.cache/huggingface/datasets
11
+
12
  CMD ["/app/.venv/bin/python", "-m", "src.infastructure.api"]