FROM corex:4.3.0

WORKDIR /workspace
COPY whls-mrv100 /packages
RUN pip install diffusers==0.34.0 sentencepiece transformers==4.55.2
# RUN pip install /packages/*.whl
COPY main.py test.sh dataset.json /workspace/