jashp2323 commited on
Commit
1dbde56
·
1 Parent(s): 42a5d4e

chore: upgrade torch to >=2.6.0 for CVE-2025-32434 compatibility

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -18,7 +18,7 @@ RUN useradd -m -u 1000 user
18
 
19
  # Install Python dependencies as root
20
  COPY requirements.txt .
21
- RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
24
  # Copy application files and change ownership to the non-root user
 
18
 
19
  # Install Python dependencies as root
20
  COPY requirements.txt .
21
+ RUN pip install --no-cache-dir "torch>=2.6.0" --index-url https://download.pytorch.org/whl/cpu
22
  RUN pip install --no-cache-dir -r requirements.txt
23
 
24
  # Copy application files and change ownership to the non-root user
requirements.txt CHANGED
@@ -2,7 +2,7 @@ fastapi>=0.100.0
2
  uvicorn>=0.20.0
3
  pillow>=9.0.0
4
  numpy>=1.20.0
5
- torch>=2.0.0
6
  transformers>=4.30.0
7
  python-multipart>=0.0.6
8
  datasets
 
2
  uvicorn>=0.20.0
3
  pillow>=9.0.0
4
  numpy>=1.20.0
5
+ torch>=2.6.0
6
  transformers>=4.30.0
7
  python-multipart>=0.0.6
8
  datasets