anfastech commited on
Commit
81fffe0
·
1 Parent(s): 8cb0e37

Initial test commit #4

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -15,8 +15,9 @@ COPY requirements.txt .
15
 
16
  # Install PyTorch and torchaudio first (CPU version)
17
  # Install the rest of requirements
18
- RUN pip install --no-cache-dir torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu121 \
19
- && pip install --no-cache-dir -r requirements.txt
 
20
 
21
  # Copy application files
22
  COPY . .
 
15
 
16
  # Install PyTorch and torchaudio first (CPU version)
17
  # Install the rest of requirements
18
+ RUN pip install --no-cache-dir torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu121
19
+
20
+ RUN pip install --no-cache-dir -r requirements.txt
21
 
22
  # Copy application files
23
  COPY . .