dataheal / Dockerfile
ravindraog's picture
Upload 3 files
93937b0 verified
FROM python:3.11-slim
# Set environment variables
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
# Create a non-root user (required by Hugging Face Spaces)
RUN useradd -m -u 1000 user
ENV HOME=/home/user
ENV PATH="/home/user/.local/bin:$PATH"
# Install system dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
&& rm -rf /var/lib/apt/lists/*
# Set working directory and ensure user owns it
WORKDIR /home/user/app
RUN chown -R user:user /home/user/app
# Create directories needed at runtime (while still root)
RUN mkdir -p experience_logs && chown -R user:user experience_logs
# Copy requirements first for better layer caching
COPY --chown=user:user requirements.txt .
# Switch to non-root user
USER user
# Install Python dependencies (CPU-only PyTorch to keep image small)
RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu && \
pip install --no-cache-dir -r requirements.txt
# Copy application code and model files
COPY --chown=user:user heal.py .
COPY --chown=user:user models/ ./models/
# Expose the Hugging Face Spaces default port
EXPOSE 7860
# Run with gunicorn for production
CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "heal:app"]