Spaces:
Runtime error
Runtime error
SoSa123456
commited on
Commit
•
ac7c4d1
1
Parent(s):
50c6ac6
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -10,7 +10,7 @@ ADD . /app
|
|
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 |
-
|
14 |
# Make port 80 available to the world outside this container
|
15 |
EXPOSE 80
|
16 |
|
|
|
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
|
16 |
|