DucThuanTran commited on
Commit
2d0380b
·
verified ·
1 Parent(s): 1c3374c

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -5
Dockerfile CHANGED
@@ -6,6 +6,13 @@ FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
6
  # Đánh dấu người tạo (tùy chọn)
7
  LABEL maintainer="your-name"
8
 
 
 
 
 
 
 
 
9
  # Cài đặt các gói hệ thống cần thiết, ví dụ: git, ffmpeg
10
  RUN apt-get update && apt-get install -y \
11
  git \
@@ -21,15 +28,13 @@ WORKDIR /app
21
  COPY requirements.txt .
22
 
23
  # Cài đặt các thư viện Python từ file requirements.txt
24
- # --no-cache-dir để giảm kích thước image
25
  RUN pip install --no-cache-dir -r requirements.txt
26
 
27
- # Sao chép toàn bộ code của ứng dụng (main.py và các file khác nếu có) vào
28
  COPY . .
29
 
30
- # Mở cổng 8000 để bên ngoài có thể truy cập vào API
31
  EXPOSE 8000
32
 
33
- # Lệnh mặc định để khởi chạy ứng dụng FastAPI khi container bắt đầu
34
- # Sử dụng uvicorn để chạy ứng dụng được định nghĩa là 'app' trong file 'main.py'
35
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
 
6
  # Đánh dấu người tạo (tùy chọn)
7
  LABEL maintainer="your-name"
8
 
9
+ # === THÊM 2 DÒNG NÀY VÀO ===
10
+ # Thiết lập biến môi trường để chỉ định thư mục cache
11
+ # Điều này sẽ lưu model vào /app/cache thay vì /.cache, nơi ứng dụng có quyền ghi
12
+ ENV TRANSFORMERS_CACHE="/app/cache"
13
+ ENV HF_HOME="/app/cache"
14
+ # ============================
15
+
16
  # Cài đặt các gói hệ thống cần thiết, ví dụ: git, ffmpeg
17
  RUN apt-get update && apt-get install -y \
18
  git \
 
28
  COPY requirements.txt .
29
 
30
  # Cài đặt các thư viện Python từ file requirements.txt
 
31
  RUN pip install --no-cache-dir -r requirements.txt
32
 
33
+ # Sao chép toàn bộ code của ứng dụng
34
  COPY . .
35
 
36
+ # Mở cổng 8000
37
  EXPOSE 8000
38
 
39
+ # Lệnh mặc định để khởi chạy ứng dụng FastAPI
 
40
  CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]