Upload Dockerfile with huggingface_hub

Dockerfile

@@ -1,19 +1,35 @@
+# Use Python 3.10 (Stable for Torch/Transformers)
 FROM python:3.10-slim
 
-WORKDIR /app
+# Set environment variables
+ENV PYTHONDONTWRITEBYTECODE 1
+ENV PYTHONUNBUFFERED 1
+ENV PORT 7860
+
+# Install system dependencies (OpenCV requires libgl1)
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
 
-# Install minimal system dependencies
-RUN apt-get update && apt-get install -y libglib2.0-0 && rm -rf /var/lib/apt/lists/*
+# Set working directory
+WORKDIR /app
 
-# Install Python requirements
+# Copy requirements and install
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy all the model files and api.py
+# Pre-download models to bake them into the Docker image
+# This ensures fast startup on Hugging Face Spaces
+RUN python -c "from transformers import TrOCRProcessor, VisionEncoderDecoderModel; \
+    TrOCRProcessor.from_pretrained('microsoft/trocr-large-handwritten'); \
+    VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-large-handwritten')"
+
+# Copy the rest of the application
 COPY . .
 
-# Expose port 7860 (Hugging Face Spaces default port)
+# Expose the HF default port
 EXPOSE 7860
 
-# Start the FastAPI server using uvicorn
-CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]
+# Command to run the application
+CMD ["python", "api.py"]