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