Spaces:
Sleeping
Sleeping
Commit
·
021d7a4
1
Parent(s):
d179ac1
Update PDF to Markdown converter API with NVIDIA L4 support
Browse files- Dockerfile +1 -1
- 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.
|
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.
|
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
|