ImageDataExtractor2 / Dockerfile
WebashalarForML's picture
Update Dockerfile
8fd8541 verified
raw
history blame
1.95 kB
# Use an official Python runtime as a parent image
FROM python:3.9-slim
# Set environment variables for Python
ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
# Set environment variables
#ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
#ENV PATH="/home/user/.local/bin:$PATH"
#ENV HF_HOME="/app/.cache/huggingface"
# Create cache directories
RUN mkdir -p /app/.cache/huggingface /app/.cache/matplotlib
# Set environment variables for Hugging Face and Matplotlib cache
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
ENV HF_HOME=/app/.cache/huggingface
ENV MPLCONFIGDIR=/app/.cache/matplotlib
# Adding permission for the cache
RUN chmod -R 777 /app/.cache
# Set the working directory
WORKDIR /app
# Install system dependencies, including libgomp
RUN apt-get update && apt-get install -y \
libgl1-mesa-glx \
libgomp1 \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Set environment variables for Huggingface and Matplotlib cache directories
ENV MPLCONFIGDIR=/tmp/matplotlib
ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
# Copy the requirements file into the container at /app
COPY requirements.txt /app/
# Install any needed packages specified in requirements.txt
RUN pip install --no-cache-dir --upgrade -r requirements.txt
# Create directories for session storage and uploads
#RUN mkdir -p /app/flask_sessions /app/uploads && chmod -R 777 /app/flask_sessions /app/uploads
# Create directories for session storage, uploads, and cache
RUN mkdir -p /app/flask_sessions /app/uploads /tmp/matplotlib /tmp/transformers_cache && chmod -R 777 /app/flask_sessions /app/uploads /tmp/matplotlib /tmp/transformers_cache
# Copy the rest of the application code to /app
COPY . /app/
# Expose the port that the app runs on
EXPOSE 7860
# Set environment variables for Flask
ENV FLASK_APP=app.py
ENV FLASK_ENV=production
# Command to run the Flask app using Gunicorn
CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860", "--timeout", "120", "app:app"]