Update PDF to Markdown converter API with NVIDIA L4 support

Dockerfile

@@ -31,12 +31,16 @@ WORKDIR /app
 # Copy requirements first
 COPY requirements.txt .
 
-# Install PyTorch dependencies first
+# Install PyTorch dependencies with specific versions known to work with transformers
+# Using torch 2.0.1 with compatible torchvision 0.15.2 which is a proven stable combination
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir torch==2.5.0+cu118 torchvision==0.20.0+cu118 torchaudio==2.5.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+    pip install --no-cache-dir torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
+
+# Ensure transformers is pinned to a compatible version
+RUN pip install --no-cache-dir transformers==4.30.2
 
 # Install remaining Python dependencies
-RUN grep -v "torch\|torchvision\|torchaudio" requirements.txt > requirements_no_torch.txt && \
+RUN grep -v "torch\|torchvision\|torchaudio\|transformers" requirements.txt > requirements_no_torch.txt && \
     pip install --no-cache-dir -r requirements_no_torch.txt
 
 # Create output directories

requirements.txt

@@ -3,6 +3,7 @@ uvicorn==0.23.2
 python-multipart==0.0.6
 marker-pdf==1.6.2
 # Torch packages are installed directly in Dockerfile
-# torch==2.5.1
+# torch==2.0.1
 # torchvision==0.15.2
-# torchaudio==2.0.2 
+# torchaudio==2.0.2
+# transformers==4.30.2