MarfinF commited on
Commit
2cb44f8
·
1 Parent(s): 1eae99a

- fix docker model marfin not found

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -4,6 +4,7 @@ WORKDIR /app
4
 
5
  COPY requirements.txt .
6
  RUN pip install --no-cache-dir -r requirements.txt
 
7
  RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='MarfinF/marfin_emotion')"
8
 
9
  COPY . .
 
4
 
5
  COPY requirements.txt .
6
  RUN pip install --no-cache-dir -r requirements.txt
7
+ RUN pip install -U transformers accelerate peft huggingface_hub
8
  RUN python -c "from transformers import pipeline; pipeline('zero-shot-classification', model='MarfinF/marfin_emotion')"
9
 
10
  COPY . .