Dockerfile

# 1. Use a base image with Python and CUDA (for the GPU)
FROM python:3.11-slim

# 2. Set the working directory
WORKDIR /code

# 3. Copy all your application files into the container
COPY . .

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

# 5. Run the fine-tuning script when the Space starts
CMD ["python", "fine_tune.py"]