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

fix deprecated params

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -4,11 +4,11 @@ FROM python:3.10-slim
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
 
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 (sử dụng HF_HOME thay vì TRANSFORMERS_CACHE)
8
+ ENV HF_HOME=/tmp/.cache
9
 
10
+ # Tải sẵn mô hình PhoBERT từ TensorFlow
11
+ RUN python -c "from transformers import AutoModel, AutoTokenizer; AutoModel.from_pretrained('vinai/phobert-base', from_tf=True); AutoTokenizer.from_pretrained('vinai/phobert-base')"
12
 
13
  # Tạo thư mục làm việc trong container
14
  WORKDIR /app