VietCat commited on
Commit
bf4d78b
·
1 Parent(s): b269ebb

update Dockerfile to fix permission issue

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -3
Dockerfile CHANGED
@@ -1,16 +1,23 @@
 
1
  FROM python:3.10-slim
2
 
3
  # Cài đặt các thư viện cần thiết
4
  RUN pip install --no-cache-dir torch transformers flask
5
 
6
- # Tạo thư mục làm việc
 
 
 
 
 
 
7
  WORKDIR /app
8
 
9
  # Copy mã nguồn vào container
10
  COPY . /app
11
 
12
- # Expose cổng 7860 (cổng mặc định HF Space dùng)
13
  EXPOSE 7860
14
 
15
- # Chạy app
16
  CMD ["python", "app.py"]
 
1
+ # Sử dụng image python slim
2
  FROM python:3.10-slim
3
 
4
  # Cài đặt các thư viện cần thiết
5
  RUN pip install --no-cache-dir torch transformers flask
6
 
7
+ # Đặt biến môi trường cho cache
8
+ ENV TRANSFORMERS_CACHE=/tmp/.cache
9
+
10
+ # Tải sẵn mô hình PhoBERT để tránh lỗi khi chạy
11
+ RUN python -c "from transformers import AutoTokenizer; AutoTokenizer.from_pretrained('vinai/phobert-base')"
12
+
13
+ # Tạo thư mục làm việc trong container
14
  WORKDIR /app
15
 
16
  # Copy mã nguồn vào container
17
  COPY . /app
18
 
19
+ # Expose cổng 7860 để chạy ứng dụng
20
  EXPOSE 7860
21
 
22
+ # Chạy ứng dụng
23
  CMD ["python", "app.py"]