train-modle / Dockerfile
fokan's picture
Force Space rebuild v2.1.0 with incremental training
0c69053
FROM python:3.9-slim
# Create a non-root user
RUN useradd --create-home --shell /bin/bash app
# Set working directory
WORKDIR /app
# Install system dependencies
RUN apt-get update && apt-get install -y \
git \
curl \
&& rm -rf /var/lib/apt/lists/*
# Copy requirements first for better caching
COPY requirements.txt .
# Install Python dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Copy application code
COPY . .
# Create necessary directories with proper permissions
RUN mkdir -p uploads models temp logs /tmp/cache && \
chown -R app:app /app /tmp/cache && \
chmod -R 755 /app
# Set environment variables
ENV PYTHONPATH=/app
ENV PORT=7860
ENV TRANSFORMERS_CACHE=/tmp/cache
ENV HF_HOME=/tmp/cache
ENV TORCH_HOME=/tmp/cache
ENV APP_VERSION=2.1.0
# Switch to non-root user
USER app
# Expose port
EXPOSE 7860
# Health check
HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
CMD curl -f http://localhost:7860/health || exit 1
# Run the application
CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]