Spaces:
Running
Running
Hardik commited on
Commit ·
f4a3f22
1
Parent(s): 1c08948
Fix Dockerfile: separate torch CPU install, add tf-keras
Browse files- 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 |
-
|
| 12 |
-
|
|
|
|
| 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 |
|