ChintanSatva commited on
Commit
440eaef
·
verified ·
1 Parent(s): e04ed9d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -31,7 +31,10 @@ WORKDIR /app
31
  # Copy requirements first for better caching
32
  COPY requirements.txt .
33
 
34
- # Install Python dependencies
 
 
 
35
  RUN pip3 install --no-cache-dir -r requirements.txt
36
 
37
  # Copy application code
 
31
  # Copy requirements first for better caching
32
  COPY requirements.txt .
33
 
34
+ # Install PyTorch CPU version first
35
+ RUN pip3 install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
36
+
37
+ # Install other Python dependencies
38
  RUN pip3 install --no-cache-dir -r requirements.txt
39
 
40
  # Copy application code