ai-matching-api / Dockerfile
quocdat1964
Init ai service
7b60154
Raw
History Blame Contribute Delete
1.9 kB
# Sử dụng base image Python 3.11 siêu nhẹ (slim)
FROM python:3.11-slim
# Cấu hình các biến môi trường cho Python
# Ngăn Python ghi file .pyc
ENV PYTHONDONTWRITEBYTECODE=1
# Bật chế độ unbuffered để log hiện ra ngay lập tức
ENV PYTHONUNBUFFERED=1
# Thư mục lưu trữ cache của HuggingFace (để dễ quản lý)
ENV HF_HOME=/app/cache
# Đặt thư mục làm việc mặc định trong container
WORKDIR /app
# Cài đặt các thư viện hệ thống cần thiết và cập nhật toàn bộ (Vá lỗi glibc)
RUN apt-get update \
&& apt-get upgrade -y \
&& apt-get install -y --no-install-recommends gcc python3-dev \
&& rm -rf /var/lib/apt/lists/*
# Copy file requirements vào container
COPY requirements.txt .
# BÍ KÍP TỐI ƯU DUNG LƯỢNG VÀ BẢO MẬT:
# 1. Cài đặt CHỈ phiên bản CPU của PyTorch (Bỏ torchvision và torchaudio vì không dùng tới xử lý ảnh/âm thanh -> Fix tận gốc lỗi Pillow)
# 2. Nâng cấp wheel để fix lỗi bảo mật của thư viện wheel
# 3. Cài các thư viện còn lại trong requirements.txt
RUN pip install --no-cache-dir --upgrade pip wheel setuptools \
&& pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu \
&& pip install --no-cache-dir -r requirements.txt
# Copy toàn bộ mã nguồn vào thư mục /app trong container
COPY . .
# Tạo thư mục cache cho HuggingFace và cấp quyền (để khi container chạy bằng user non-root có thể tải model)
RUN mkdir -p /app/cache && chmod 777 /app/cache
# Mở port mặc định ra bên ngoài
EXPOSE 8000
EXPOSE 7860
# Lệnh mặc định khi container khởi chạy
# Đổi mặc định sang 7860 để HuggingFace Spaces không bị treo ở trạng thái "Starting"
CMD ["sh", "-c", "uvicorn app.main:app --host 0.0.0.0 --port ${PORT:-7860}"]