marcosremar2 commited on
Commit
021d7a4
·
1 Parent(s): d179ac1

Update PDF to Markdown converter API with NVIDIA L4 support

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -34,7 +34,7 @@ COPY requirements.txt .
34
  # Install Python dependencies
35
  RUN pip install --no-cache-dir --upgrade pip && \
36
  pip install --no-cache-dir -r requirements.txt && \
37
- pip install --no-cache-dir torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
38
 
39
  # Create output directories
40
  RUN mkdir -p /app/output/images && \
 
34
  # Install Python dependencies
35
  RUN pip install --no-cache-dir --upgrade pip && \
36
  pip install --no-cache-dir -r requirements.txt && \
37
+ pip install --no-cache-dir torch==2.5.1+cu118 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
38
 
39
  # Create output directories
40
  RUN mkdir -p /app/output/images && \
requirements.txt CHANGED
@@ -2,6 +2,6 @@ fastapi==0.104.1
2
  uvicorn==0.23.2
3
  python-multipart==0.0.6
4
  marker-pdf==1.2.4
5
- torch==2.0.1
6
  torchvision==0.15.2
7
  torchaudio==2.0.2
 
2
  uvicorn==0.23.2
3
  python-multipart==0.0.6
4
  marker-pdf==1.2.4
5
+ torch==2.5.1
6
  torchvision==0.15.2
7
  torchaudio==2.0.2