VietCat commited on
Commit
cfc05e8
·
1 Parent(s): fc07040

update lib

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -8,7 +8,7 @@ RUN pip install --no-cache-dir transformers flask tensorflow tf-keras
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 TFAutoModel, AutoTokenizer; TFAutoModel.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
 
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 TFAutoModel, AutoTokenizer; TFAutoModel.from_pretrained('vinai/phobert-base'); AutoTokenizer.from_pretrained('vinai/phobert-base')"
12
 
13
  # Tạo thư mục làm việc trong container
14
  WORKDIR /app