Princess3 commited on
Commit
c9fff6a
1 Parent(s): 1c0c554

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -20,7 +20,7 @@ WORKDIR /app
20
  # Install Python dependencies
21
  RUN pip install --upgrade pip
22
  RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
23
- RUN pip install transformers accelerate requests psutil termcolor sklearn
24
 
25
  # Copy the current directory contents into the container at /app
26
  COPY . /app
 
20
  # Install Python dependencies
21
  RUN pip install --upgrade pip
22
  RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
23
+ RUN pip install transformers accelerate requests psutil termcolor sk-learn
24
 
25
  # Copy the current directory contents into the container at /app
26
  COPY . /app