B4869 commited on
Commit
3dc2abf
·
1 Parent(s): bcd6c8a

bug fix again

Browse files
Files changed (2) hide show
  1. Dockerfile +4 -0
  2. download_model.py +4 -0
Dockerfile CHANGED
@@ -12,6 +12,10 @@ ENV PATH="/home/user/.local/bin:$PATH"
12
  # ใช้ /tmp เพราะมี space มากกว่า default location (~/.cache/)
13
  ENV HF_HOME="/home/user/.cache/huggingface"
14
 
 
 
 
 
15
  WORKDIR /app
16
 
17
  # Copy requirements และ install dependencies
 
12
  # ใช้ /tmp เพราะมี space มากกว่า default location (~/.cache/)
13
  ENV HF_HOME="/home/user/.cache/huggingface"
14
 
15
+ # รับ HF_TOKEN จาก build args สำหรับ gated models
16
+ ARG HF_TOKEN
17
+ ENV HF_TOKEN=$HF_TOKEN
18
+
19
  WORKDIR /app
20
 
21
  # Copy requirements และ install dependencies
download_model.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  from sentence_transformers import SentenceTransformer
2
 
3
  print("Downloading model...")
 
1
+ """
2
+ Script สำหรับ download model ล่วงหน้าตอน Docker build
3
+ เพื่อ cache model ไว้ใน Docker image
4
+ """
5
  from sentence_transformers import SentenceTransformer
6
 
7
  print("Downloading model...")