Spaces:
Running
Running
update lib
Browse files- 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'
|
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
|