Spaces:
Running
Running
fix deprecated params
Browse files- 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
|
9 |
|
10 |
-
# Tải sẵn mô hình PhoBERT
|
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
|