MarfinF commited on
Commit
47bc4fb
·
1 Parent(s): f4ac0cd

- update docker to download marfin_emotion model

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -2,8 +2,10 @@ FROM python:3.9-slim
2
 
3
  WORKDIR /app
4
 
 
 
 
5
  COPY . .
6
 
7
- RUN pip install -r requirements.txt
8
 
9
  CMD ["uvicorn", "backend.app:app", "--host", "0.0.0.0", "--port", "7860"]
 
2
 
3
  WORKDIR /app
4
 
5
+ RUN pip install --no-cache-dir -r requirements.txt
6
+ RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='MarfinF/marfin_emotion')"
7
+
8
  COPY . .
9
 
 
10
 
11
  CMD ["uvicorn", "backend.app:app", "--host", "0.0.0.0", "--port", "7860"]