Hardik commited on
Commit
f4a3f22
·
1 Parent(s): 1c08948

Fix Dockerfile: separate torch CPU install, add tf-keras

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -8,8 +8,9 @@ RUN apt-get update && apt-get install -y curl nginx supervisor && \
8
  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
- torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu \
12
- transformers==4.41.2 joblib==1.4.2 tf-keras
 
13
 
14
  RUN python -c "import nltk; nltk.download('stopwords', quiet=True); nltk.download('punkt', quiet=True); nltk.download('wordnet', quiet=True)"
15
 
 
8
  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
 
15
  RUN python -c "import nltk; nltk.download('stopwords', quiet=True); nltk.download('punkt', quiet=True); nltk.download('wordnet', quiet=True)"
16