SoSa123456 commited on
Commit
e350155
1 Parent(s): c296645

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -6,10 +6,13 @@ WORKDIR /app
6
 
7
  # Copy the current directory contents into the container at /app
8
  ADD . /app
9
-
 
10
  # Install any needed packages specified in requirements.txt
11
  RUN python -m pip install --upgrade pip
 
12
  RUN pip install -r requirements.txt
 
13
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
14
  # Make port 80 available to the world outside this container
15
  #EXPOSE 80
 
6
 
7
  # Copy the current directory contents into the container at /app
8
  ADD . /app
9
+ RUN sudo apt-get update -y
10
+ RUN sudo apt-get upgrade -y
11
  # Install any needed packages specified in requirements.txt
12
  RUN python -m pip install --upgrade pip
13
+ RUN pip install transformers[torch]
14
  RUN pip install -r requirements.txt
15
+
16
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
17
  # Make port 80 available to the world outside this container
18
  #EXPOSE 80