B4869 commited on
Commit
908160b
·
1 Parent(s): 3a3f3a9

edit dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -10
  2. download_model.py +0 -4
Dockerfile CHANGED
@@ -8,27 +8,22 @@ RUN useradd -m -u 1000 user
8
  USER user
9
  ENV PATH="/home/user/.local/bin:$PATH"
10
 
11
- # ตั้งค่า cache directory สำหรับ Hugging Face models
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
22
  COPY --chown=user ./requirements.txt requirements.txt
23
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
24
 
 
25
  COPY --chown=user ./download_model.py download_model.py
26
 
27
  # Copy application code
28
  COPY --chown=user . /app
29
 
30
- # Expose port 7860 (default port สำหรับ Hugging Face Spaces)
31
  EXPOSE 7860
32
 
33
- # Run Flask app with Gunicorn (production WSGI server)
34
- CMD ["/bin/bash", "-c", "python download_model.py && gunicorn app:app --bind 0.0.0.0:7860 --workers 1 --timeout 120"]
 
8
  USER user
9
  ENV PATH="/home/user/.local/bin:$PATH"
10
 
11
+ # ตั้งค่า cache directory
 
12
  ENV HF_HOME="/home/user/.cache/huggingface"
13
 
 
 
 
 
14
  WORKDIR /app
15
 
16
+ # Copy requirements และ install
17
  COPY --chown=user ./requirements.txt requirements.txt
18
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
19
 
20
+ # Copy script (แต่ยังไม่รัน รอไปรันใน CMD)
21
  COPY --chown=user ./download_model.py download_model.py
22
 
23
  # Copy application code
24
  COPY --chown=user . /app
25
 
 
26
  EXPOSE 7860
27
 
28
+ # CMD นี้จะทำงานตอน Runtime ซึ่ง ตอนนั้นจะมี HF_TOKEN ให้ใช้แล้ว
29
+ CMD ["/bin/bash", "-c", "python download_model.py && gunicorn app:app --bind 0.0.0.0:7860 --workers 1 --timeout 120"]
download_model.py CHANGED
@@ -1,7 +1,3 @@
1
- """
2
- Script สำหรับ download model ล่วงหน้าตอน Docker build
3
- เพื่อ cache model ไว้ใน Docker image
4
- """
5
  #login huggingface
6
  from huggingface_hub import login
7
  import os
 
 
 
 
 
1
  #login huggingface
2
  from huggingface_hub import login
3
  import os