Update Dockerfile

Dockerfile

@@ -1,31 +1,16 @@
-# Use an official Python runtime as a parent image
-FROM python:3.10-slim
+# Use the official lightweight CPU-only image from Docling
+FROM quay.io/docling-project/docling-serve-cpu:latest
 
-# Set the working directory
-WORKDIR /app
+# Set environment variables for Hugging Face Spaces
+ENV HOST=0.0.0.0
+ENV PORT=7860
+ENV DOCLING_SERVE_ENABLE_UI=1
 
-# Install system dependencies required by PaddleOCR and OpenCV
-# ADDED: libgomp1
-RUN apt-get update && apt-get install -y \
-    libgl1 \
-    libglib2.0-0 \
-    libgomp1 \
-    && rm -rf /var/lib/apt/lists/*
+# Pre-download the SmolDocling model to speed up startup (optional but recommended)
+# This ensures the 256M model is ready in the container cache
+RUN python3 -c "from docling.document_converter import DocumentConverter; DocumentConverter()"
 
-# Create a non-root user required by Hugging Face Spaces
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-# Copy requirements and install
-COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy the rest of the application
-COPY --chown=user . .
-
-# Expose the port FastAPI runs on
 EXPOSE 7860
 
-# Command to run the application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Start the server
+CMD ["docling-serve", "run", "--host", "0.0.0.0", "--port", "7860"]