Spaces:
Sleeping
Sleeping
edit dockerfile
Browse files- Dockerfile +5 -10
- 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
|
| 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
|
| 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 |
-
#
|
| 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
|