Spaces:
Runtime error
Runtime error
SoSa123456
commited on
Commit
•
e350155
1
Parent(s):
c296645
Update Dockerfile
Browse files- 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
|