viet-toxic-classifier / Dockerfile
dongtruong1910
Deploy
f817340
raw
history blame contribute delete
864 Bytes
# 1. Chọn hệ điều hành Python 3.9
FROM python:3.9
# 2. Tạo thư mục làm việc
WORKDIR /code
# 3. Copy file requirements và cài đặt thư viện
COPY ./requirements.txt /code/requirements.txt
# Cài torch bản CPU cho nhẹ (tùy chọn, hoặc cài thường cũng được)
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
# 4. Copy toàn bộ code vào trong
COPY ./src /code/src
COPY ./saved_models /code/saved_models
COPY ./api.py /code/api.py
# 5. Cấp quyền cho user (Hugging Face yêu cầu)
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# 6. Mở cổng 7860 (Cổng bắt buộc của Hugging Face)
# Lưu ý: Code api.py của bạn đang chạy port 8000, ta sẽ đổi lệnh chạy ở đây
CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]