# Use Python 3.10-slim as the base image
FROM python:3.10-slim

# Set Hugging Face cache dir to a writable path
ENV HF_HOME=/tmp/hf_cache

# Create the cache directory and give full permissions
RUN mkdir -p /tmp/hf_cache && chmod -R 777 /tmp/hf_cache

# Set working directory
WORKDIR /code

# Copy files to the container
COPY . .

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

# Run training script
CMD ["python", "train.py"]