From 693d64e5e3200371998126c25e3f86a6e87d113a Mon Sep 17 00:00:00 2001
From: sergio <sergiojj932@gmail.com>
Date: Sat, 17 Jan 2026 11:22:38 +0100
Subject: [PATCH] gpu fix

---
 src/paddle_ocr/Dockerfile.gpu | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/src/paddle_ocr/Dockerfile.gpu b/src/paddle_ocr/Dockerfile.gpu
index 124295a..6039933 100644
--- a/src/paddle_ocr/Dockerfile.gpu
+++ b/src/paddle_ocr/Dockerfile.gpu
@@ -51,12 +51,18 @@ RUN if ls /tmp/wheels/paddlepaddle*.whl 1>/dev/null 2>&1; then \
     else \
         echo "=== Installing PaddlePaddle from CUDA index (x86_64) ===" && \
         pip install --no-cache-dir paddlepaddle-gpu==3.2.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/; \
-    fi
+    fi && \
+    rm -rf /tmp/wheels
 
-# Install remaining dependencies (skip paddlepaddle-gpu line from requirements)
-RUN grep -v "paddlepaddle-gpu" requirements-gpu.txt > /tmp/requirements-no-paddle.txt && \
-    pip install --no-cache-dir -r /tmp/requirements-no-paddle.txt && \
-    rm -rf /tmp/wheels /tmp/requirements-no-paddle.txt
+# Install remaining dependencies explicitly
+RUN pip install --no-cache-dir \
+    paddleocr==3.3.2 \
+    jiwer \
+    numpy \
+    fastapi \
+    "uvicorn[standard]" \
+    pydantic \
+    Pillow
 
 # Copy application code
 COPY paddle_ocr_tuning_rest.py .