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

update lib

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -2,7 +2,7 @@
2
  FROM python:3.10-slim
3
 
4
  # Cài đặt các thư viện cần thiết
5
- RUN pip install --no-cache-dir transformers flask tensorflow
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
 
2
  FROM python:3.10-slim
3
 
4
  # Cài đặt các thư viện cần thiết
5
+ RUN pip install --no-cache-dir transformers flask tensorflow tf-keras
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