# Start from a PyTorch + CUDA base image
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel

WORKDIR /code
COPY . .

# Install dependencies
RUN pip install -r requirements.txt
RUN pip install flash-attn --no-build-isolation

# Make the startup script executable
COPY startup.sh .
RUN chmod +x startup.sh

# This is the new command: it runs your script automatically when the Space starts.
CMD ["./startup.sh"]