Hardik commited on
Commit
eafed38
·
1 Parent(s): 7772b12

Docker: remove tf-keras, keep native keras 3

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -3
Dockerfile CHANGED
@@ -9,9 +9,7 @@ RUN pip install --no-cache-dir fastapi==0.111.0 uvicorn==0.30.1 pydantic==2.7.4
9
  pydantic-settings==2.3.4 numpy==1.26.4 pandas==2.2.2 scikit-learn==1.3.2 \
10
  scipy==1.13.1 nltk==3.8.1 tensorflow==2.16.1 keras==3.3.3 \
11
  transformers==4.41.2 joblib==1.4.2 && \
12
- pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu && \
13
- pip install --no-cache-dir tf-keras && \
14
- python -c "import tf_keras; print('tf-keras OK')"
15
 
16
  RUN python -c "import nltk; nltk.download('stopwords', quiet=True); nltk.download('punkt', quiet=True); nltk.download('wordnet', quiet=True)"
17
 
 
9
  pydantic-settings==2.3.4 numpy==1.26.4 pandas==2.2.2 scikit-learn==1.3.2 \
10
  scipy==1.13.1 nltk==3.8.1 tensorflow==2.16.1 keras==3.3.3 \
11
  transformers==4.41.2 joblib==1.4.2 && \
12
+ pip install --no-cache-dir torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
 
 
13
 
14
  RUN python -c "import nltk; nltk.download('stopwords', quiet=True); nltk.download('punkt', quiet=True); nltk.download('wordnet', quiet=True)"
15