Update Dockerfile

Dockerfile

@@ -1,15 +1,20 @@
 # Use the official Hugging Face Spaces GPU image
 FROM huggingface/zero-gpu-gradio:latest
 
-# Build llama-cpp-python with CUDA support for the RTX Pro 6000 Blackwell
-ENV CMAKE_ARGS="-DGGML_CUDA=on"
-ENV FORCE_CMAKE=1
-
-# Install the Python dependencies (this triggers the llama-cpp-python CUDA build)
+# --- Install Python dependencies ---
 COPY requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
-# Copy the app
+# --- Install llama-cpp-python with CUDA 12.4 prebuilt wheel ---
+# (prebuilt.sh never runs when a custom Dockerfile exists,
+#  so we install it here in the Docker build step instead)
+RUN pip install --no-cache-dir llama-cpp-python \
+  --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu124
+
+# --- File parsers ---
+RUN pip install --no-cache-dir python-docx pdfplumber chardet
+
+# --- Copy app ---
 COPY app.py /home/user/app/app.py
 
 CMD ["python", "app.py"]