Spaces:
Runtime error
Runtime error
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
|