VietCat commited on
Commit
8ef2756
·
1 Parent(s): 502ff6b

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
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
 
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