File size: 1,223 Bytes
ada534f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# Lightweight Dockerfile for HuggingFace Spaces
FROM python:3.10-slim
WORKDIR /app
# Install minimal system dependencies
RUN apt-get update && apt-get install -y \
libgl1-mesa-glx \
libglib2.0-0 \
curl \
&& rm -rf /var/lib/apt/lists/* \
&& apt-get clean
# Copy and install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir --upgrade pip \
&& pip install --no-cache-dir -r requirements.txt \
&& pip cache purge
# Copy application files (models excluded via .dockerignore)
COPY main.py .
COPY download_models.py .
COPY README.md .
# Create necessary directories
RUN mkdir -p models uploads
# Download models at runtime (smaller build)
# Models will be downloaded when the app starts
# Set environment variables for HuggingFace Spaces
ENV PYTHONUNBUFFERED=1
ENV GRADIO_SERVER_NAME=0.0.0.0
ENV GRADIO_SERVER_PORT=7860
# Expose HuggingFace Spaces port
EXPOSE 7860
# Health check
HEALTHCHECK --interval=30s --timeout=10s --start-period=120s --retries=3 \
CMD curl -f http://localhost:7860/health || exit 1
# Run with model download and reduced memory usage
CMD ["sh", "-c", "python download_models.py && uvicorn main:app --host 0.0.0.0 --port 7860 --workers 1"]
|