Dockerfile

FROM python:3.10

# Set working directory
WORKDIR /app

# Copy files
COPY . .

# Install dependencies
RUN pip install -r requirements.txt

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