Dockerfile

# Use an official Python runtime
FROM python:3.9-slim

# Set the working directory
WORKDIR /app

# Copy the requirements file
COPY requirements.txt .

# Change permission
RUN mkdir -p /cache/huggingface /cache/matplotlib \
    && chmod -R 777 /cache/huggingface /cache/matplotlib

# Set environment variables for Hugging Face and Matplotlib 
ENV HF_HOME=/cache/huggingface
ENV MPLCONFIGDIR=/cache/matplotlib

# Flask
ENV FLASK_APP=app.py

# Install dependencies
RUN pip install --no-cache-dir -r requirements.txt

RUN "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface'); AutoTokenizer.from_pretrained('gpt2-xl', cache_dir='/cache/huggingface')"

# Copy the rest of the app
COPY . .

# Expose the Flask port
EXPOSE 7860

# Run the application
CMD ["flask", "run", "--host=0.0.0.0", "--port=7860"]