Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -29,7 +29,7 @@ COPY . /app
|
|
29 |
# Install Python dependencies
|
30 |
RUN pip install --no-cache-dir -U pip setuptools wheel
|
31 |
RUN pip install --no-cache-dir setuptools-rust
|
32 |
-
|
33 |
RUN pip install --no-cache-dir -r requirements.txt
|
34 |
|
35 |
# Create and switch to a non-root user
|
|
|
29 |
# Install Python dependencies
|
30 |
RUN pip install --no-cache-dir -U pip setuptools wheel
|
31 |
RUN pip install --no-cache-dir setuptools-rust
|
32 |
+
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
33 |
RUN pip install --no-cache-dir -r requirements.txt
|
34 |
|
35 |
# Create and switch to a non-root user
|