FROM aisiuk/inspect-tool-support

RUN mkdir -p /workspace
WORKDIR /workspace

RUN mkdir -p /utils
COPY eval_data/reference_gradcam.jpg /utils/reference_gradcam.jpg
COPY eval_data/reference_gradients.pt /utils/reference_gradients.pt
COPY eval_data/reference_activations.pt /utils/reference_activations.pt

RUN pip install torch==2.5.1
